Pytorch模型保存和加载

torch训练好一个模型，如何保存模型和参数？

• 想要保存整个模型，命令
  torch.save(model,'path/name.pt')
• 仅保存模型参数
  torch.save(model.state_dict(), 'path/name.pt')

那么，如何加载模型呢？
model.load_state_dict(torch.load("path/name.pt"))

Remark:
在评估模式的时候调用model.eval()来固定dropout和批次归一化。否则会产生不一致的推理结果。