模型保存

• 模型保存
  • 基本

基本

• pytorch可以保存模型的参数，后续部署时可以实例化模型后读取参数，示例：

net = MyNet()
# 保存
torch.save(net.state_dict(), 'MyNet')
# 加载
clone = MyNet() # 先实例化一个模型实例
clone.load_state_dict(torch.load('MyNet'))