AttributeError: '_IncompatibleKe

2021-12-14  本文已影响0人  1037号森林里一段干木头

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()
#保存的权重文件
resNet50 = resnet.resnet50()
resNet50.load_state_dict(torch.load(resNet_path)['state_dict'])
resNet50.eval()
上一篇 下一篇

猜你喜欢

热点阅读