module 'torch' has no attribute 'load_state_dict'
时间: 2023-08-04 22:05:45 浏览: 482
pytorch 状态字典:state_dict使用详解
5星 · 资源好评率100%
根据您提供的引用内容,出现"module 'torch' has no attribute 'load_state_dict'"的错误是因为在较旧版本的PyTorch中,torch模块没有load_state_dict()方法。[2]为了解决这个问题,您可以尝试使用更高版本的PyTorch,如1.6及以上版本。[3]在1.6版本之后,PyTorch默认使用zip文件格式来保存权重文件,而较旧版本的PyTorch无法直接加载这些权重文件。您可以在新建的1.6版本的PyTorch环境中运行以下代码来解决这个问题:
```python
import torch
state_dict = torch.load("faster_rcnn_orpn_r101_fpn_1x_mssplit_rr_dota10_epoch12.pth", map_location='cpu')
torch.save(state_dict, "faster_rcnn_orpn_r101_fpn_1x_mssplit_rr_dota10_epoch12.pth", _use_new_zipfile_serialization=False)
```
这样可以将权重文件保存为旧版本可以加载的格式,然后您可以尝试使用load_state_dict()方法加载权重文件。[1]请确保您已经正确安装了1.6版本的PyTorch,并按照上述步骤进行操作。
阅读全文