加载模型不需要评估？

Question

在加载模型后尝试使用测试

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

但这会吐出错误

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

Answer 1

这里您不需要将net.load_state_dict分配给net

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

刚刚使用

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

更多信息，请参见Recommended approach for saving a model