torch.save之后的报错

来源:4-4 利用神经网络解决分类和回归问题(2)

weixin_慕尼黑0465866

2021-09-26

Traceback (most recent call last):
File “C:\Users\ybbin.conda\envs\pytorch1\lib\site-packages\IPython\core\interactiveshell.py”, line 3441, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File “”, line 1, in
torch.save(net,“model/model.pkl”) # 将模型整体性保存
File “C:\Users\ybbin.conda\envs\pytorch1\lib\site-packages\torch\serialization.py”, line 224, in save
return _with_file_like(f, “wb”, lambda f: _save(obj, f, pickle_module, pickle_protocol))
File “C:\Users\ybbin.conda\envs\pytorch1\lib\site-packages\torch\serialization.py”, line 149, in _with_file_like
return body(f)
File “C:\Users\ybbin.conda\envs\pytorch1\lib\site-packages\torch\serialization.py”, line 224, in
return _with_file_like(f, “wb”, lambda f: _save(obj, f, pickle_module, pickle_protocol))
File “C:\Users\ybbin.conda\envs\pytorch1\lib\site-packages\torch\serialization.py”, line 296, in _save
pickler.dump(obj)
_pickle.PicklingError: Can’t pickle <class ‘main.Net’>: attribute lookup Net on main failed

写回答

1回答

会写代码的好厨师

2021-10-11

这个问题需要看到完整代码。另外,看下项目里面是否有model这个文件夹

0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程