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这个文件夹
00
相似问题