pytorch讀取權重文件后出現
File "test.py", line 61, in main
resNet50.eval()
AttributeError: '_IncompatibleKeys' object has no attribute 'eval'
- 代碼
#保存的權重文件
resNet50 = resnet.resnet50()
resNet50 = resNet50.load_state_dict(torch.load(resNet_path)['state_dict'])
resNet50.eval()
- 修改
這里的模型不能用“=”來賦值，第一行代碼導入模型的結構文件，下面應該直接在此模型上調用load_state_dict函數加載權重文件
#保存的權重文件
resNet50 = resnet.resnet50()
resNet50.load_state_dict(torch.load(resNet_path)['state_dict'])
resNet50.eval()
