6.30 torch.save() / ps -ef | grep / kill -9 UID_torch.save() 被 kill-CSDN博客

本文链接：https://blog.csdn.net/IsayIwant/article/details/131489108

文章介绍了使用torch.save()保存PyTorch模型时，会保存整个模型对象而非仅状态字典。在加载时，需要通过state_dict进行操作。同时，文章提到了进程控制的相关命令，如使用grep和kill命令来管理和结束进程。

摘要由CSDN通过智能技术生成

如果使用的是torch.save(model)保存模型，那么加载时确实会得到一个类对象而不是字典对象。
这是因为torch.save()函数会将整个模型对象保存到文件中，而不仅仅是模型的状态字典。

if args.resume:
    loaded_model = torch.load(args.resume, map_location=torch.device('cpu'))
    state_dict = loaded_model.state_dict()
    modified_state_dict = {f"module.{k}": v for k, v in state_dict.items()}
    model.load_state_dict(modified_state_dict)

Ctrl+c 停止进程， Ctrl+z 暂停进程
命令：ps -ef | grep 进程名称
"ps -ef | grep" 的意思是，在所有正在运行的进程中，通过管道将输出传递给grep命令，然后使用grep命令在输出中查找匹配的模式输出到终端。
ps:process status -ef（parameter）：The -ef option is used to display information about all running processes in a full format.
把所有jiahao相关的进程全部杀死
ps -ef | grep 'jiahao_+' | awk '{print $2}' | xargs kill -9

kill -9+UID号强制删除进程
kill命令用于向进程发送信号，而-9是一个特殊的信号编号，表示强制终止进程。当你执行kill -9 PID命令时，操作系统会发送一个SIGKILL信号给指定PID的进程，强制终止它的执行。

"it/s" 是指每秒迭代数（iterations per second）