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 12:04:47 浏览: 190
linux 下 pip3 报错“ File “/usr/bin/pip3”, line 9, in from pip import mainImportError: canno”处理
5星 · 资源好评率100%
这个错误是由于在保存模型时使用了`model.state.dict()`而导致的。根据错误提示,你的模型是`mylenet5`对象,并且该对象没有`state`属性。
如果你想保存模型的状态字典,你应该使用`model.state_dict()`而不是`model.state.dict()`。请修改代码如下:
```python
torch.save(model.state_dict(), 'save_model/best_model.pth')
```
这样就能正确保存模型的状态字典了。
阅读全文