AttributeError: '_IncompatibleKeys' object has no attribute 'to'
时间: 2023-08-04 11:04:53 浏览: 228
解决运行出现dict object has no attribute has_key问题
5星 · 资源好评率100%
根据引用[1]和引用[2]的内容,出现AttributeError: '_IncompatibleKeys' object has no attribute 'to'的错误是因为在加载权重文件时使用了错误的赋值方式。正确的做法是直接在模型上调用load_state_dict函数加载权重文件,而不是使用赋值等于号。所以代码应该修改为:
```
resNet50 = resnet.resnet50()
resNet50.load_state_dict(torch.load(resNet_path)['state_dict'])
resNet50.eval()
```
这样就可以避免出现AttributeError: '_IncompatibleKeys' object has no attribute 'to'的错误了。
阅读全文