Traceback (most recent call last): File "D:\PycharmProjects\pythonProject2\main.py", line 43, in <module> models.eval() AttributeError: module 'torchvision.models' has no attribute 'eval'
时间: 2024-03-25 18:41:20 浏览: 104
这个错误是因为你使用了torchvision.models.eval(),而这个eval()函数是用于模型的评估的,而不是用于模型的加载。正确的使用方式是先加载模型,然后再用模型进行评估。你可以尝试修改代码,将torchvision.models.eval()替换为加载你所使用的模型的代码。例如:
```python
import torch
model = torch.load('model.pth')
model.eval()
```
其中,model.pth是你训练好的模型的保存路径,需要根据你的实际情况进行修改。
阅读全文
相关推荐









