我想使用 torch.save() 保存权重。
我也遇到了这个问题
我的代码是这样的 文件 home/4DenoiseNet/networks/train/tasks/semantic/../../tasks/semantic/modules/trainer.py”,第 283 行,训练中 torch.save(state, self.log + "/FourDenoiseNet" + "")
我非常感谢您的帮助。
我改变了变量
像这样:
状态=无 这使得代码可以工作,但不保存
self.log = 无 运行代码不起作用
我的错误信息
回溯(最近一次调用最后一次): 文件“./train.py”,第 163 行,位于 训练师.train() 文件“/home/jsw/4DenoiseNet/networks/train/tasks/semantic/../../tasks/semantic/modules/trainer.py”,第 283 行,训练中 torch.save(状态, self.log + "/FourDenoiseNet" + "") 文件“/home/jsw/anaconda3/envs/cvpr/lib/python3.8/site-packages/torch/serialization.py”,第 441 行,保存 _save(obj,打开的zip文件,pickle_模块,pickle_协议) 文件“/home/jsw/anaconda3/envs/cvpr/lib/python3.8/site-packages/torch/serialization.py”,第 653 行,在 _save 中 pickler.dump(obj) 类型错误:无法腌制“WeakMethod”对象