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 22:04:47 浏览: 208
这个错误是由于在保存模型时使用了`model.state.dict()`而导致的。根据错误提示,你的模型是`mylenet5`对象,并且该对象没有`state`属性。
如果你想保存模型的状态字典,你应该使用`model.state_dict()`而不是`model.state.dict()`。请修改代码如下:
```python
torch.save(model.state_dict(), 'save_model/best_model.pth')
```
这样就能正确保存模型的状态字典了。
相关问题
Traceback (most recent call last): File "/home/yangqh/yolov7-main/train_.py", line 667, in <module> train(hyp, opt, device, tb_writer) File "/home/yangqh/yolov7-main/train_.py", line 310, in train torch.save(model, wdir / 'init.pt') File "/h
抱歉,这似乎是一个代码错误。根据错误信息,程序试图保存一个模型,但在保存时出现了问题。具体地说,它在执行 "torch.save(model, wdir / 'init.pt')" 这一行代码时遇到了问题。这可能是由于文件路径不正确、磁盘空间已满或其他原因导致的。您可能需要检查代码中的文件路径和磁盘空间,并确保它们正确无误。如果问题仍然存在,请考虑咨询相关人员或在开发人员社区中寻求帮助。
Traceback (most recent call last): File "E:/403/myworld/graphBatch.py", line 38, in <module> dgl.save_graphs(OUT_PATH + "merged_graph_train.bin",[merged_graph_train],'coo',{}) File "D:\code\myworld\lib\site-packages\dgl\data\graph_serialize.py", l
这个错误提示表明在执行`dgl.save_graphs`函数时,传入的参数有误。具体而言,可能是以下几个原因:
1. `OUT_PATH`变量未定义或定义有误。
2. `merged_graph_train`变量未定义或定义有误。
3. 存储格式参数`format`有误。
4. `labels`参数有误。
需要检查以上几个原因,并根据实际情况进行修改。
修改后的代码示例如下:
```python
import dgl
import torch
OUT_PATH = '/path/to/your/output/'
merged_graph_train = dgl.batch([g1, g2, g3])
# 以COO格式存储
dgl.save_graphs(OUT_PATH + "merged_graph_train.bin", [merged_graph_train], 'coo', {})
# 读取存储的图
glist, _ = dgl.load_graphs(OUT_PATH + "merged_graph_train.bin")
print(glist)
```
在上面的示例中,我们首先定义了`OUT_PATH`和`merged_graph_train`变量,并将三张图`g1`、`g2`、`g3`批量合并为`merged_graph_train`,然后以COO格式存储了`merged_graph_train`,最后用`load_graphs`函数读取了存储的图。
阅读全文