Traceback (most recent call last): File "E:\pythonProject\train.py", line 90, in <module> torch.save(model.state.dict(), 'save_model/best_model.pth') ^^^^^^^^^^^ File "E:\pythonProject\venv\Lib\site-packages\torch\nn\modules\module.py", line 1614, in __getattr__ raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'mylenet5' object has no attribute 'state'
时间: 2023-12-04 19:04:47 浏览: 54
linux 下 pip3 报错“ File “/usr/bin/pip3”, line 9, in from pip import mainImportError: canno”处理
5星 · 资源好评率100%
这个错误是由于在保存模型时使用了错误的属性名称 "state.dict()" 导致的。正确的属性名称应该是 "state_dict()"。你可以将这行代码修改为以下内容:
```python
torch.save(model.state_dict(), 'save_model/best_model.pth')
```
这样就可以正常保存模型了。
阅读全文