将.pth.tar文件转成onnx时出现错误AttributeError: 'dict' object has no attribute 'eval'
时间: 2023-08-04 20:09:08 浏览: 278
根据引用\[1\]中的信息,load_state_dict()方法返回的是_IncompatibleKeys对象,而不是模型本身。因此,在使用赋值等号时,model并不是你想要的那个模型,所以没有to、eval等属性。
根据引用\[2\]中的信息,将.pth模型转换为onnx的代码示例中,使用了torch.load()方法加载了.pth模型,并将其赋值给了model变量。然后使用torch.onnx.export()方法将模型导出为onnx格式。这种方式可以正确地将.pth模型转换为onnx格式。
根据引用\[3\]中的信息,如果你想加载保存的网络模型,可以直接使用torch.load()方法加载模型文件,并将其赋值给model变量。然后调用model.eval()方法将模型设置为评估模式。
综上所述,出现错误AttributeError: 'dict' object has no attribute 'eval'的原因是你使用了错误的方式加载模型。正确的方式是使用torch.load()方法加载模型文件,并调用model.eval()方法将模型设置为评估模式。
#### 引用[.reference_title]
- *1* [(完全解决)AttributeError: ‘_IncompatibleKeys‘ object has no attribute ‘to‘/‘eval](https://blog.csdn.net/qq_43391414/article/details/124074124)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* [Pytorch模型转ONNX](https://blog.csdn.net/yangpan011/article/details/125848508)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [pytorch载入模型时出现错误AttributeError Generator object has no attribute copy](https://blog.csdn.net/LLLLUFFY/article/details/106561536)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文