pytorch - 是加载模型不需要评估？

Question

加载模型后尝试使用测试

net = net.load_state_dict(torch.load(PATH))
net.eval()

但这吐出了错误

net.eval() AttributeError: '_IncompatibleKeys' object has no attribute 'eval'

标签： pytorch

Answer 1

在这里你不需要分配net.load_state_dict给net

net = net.load_state_dict(torch.load(PATH))

刚用过

net.load_state_dict(torch.load(PATH))
net.eval()

更多请参阅保存模型的推荐方法