码迷,mamicode.com
首页 > 其他好文 > 详细

torch保存加载模型

时间:2019-10-04 11:50:32      阅读:78      评论:0      收藏:0      [点我收藏+]

标签:code   保存加载   model   dict   ORC   load   加载   ram   stat   

  • 保存模型

    torch.save(my_model.state_dict(), "params.pkl")
  • 加载模型

    先初始化model网络结构
    model.load_state_dict(torch.load("params.pkl"))

torch保存加载模型

标签:code   保存加载   model   dict   ORC   load   加载   ram   stat   

原文地址:https://www.cnblogs.com/rise0111/p/11621640.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!