会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
贝壳里的星海
博客园
首页
新随笔
联系
订阅
管理
2021年6月23日
pytorch-模型保存和加载
摘要: pytorch-模型保存和加载 加载模型参数和选择是由保存的模型数据结构决定,故先要确定保存模型模型的方法和数据结构 保存模型 # 模型权重参数 model.state_dict() '''首先说一下 model.state_dict() pytorch 中的 model.state_dict 是一
阅读全文
posted @ 2021-06-23 21:20 贝壳里的星海
阅读(1096)
评论(0)
推荐(0)
编辑
公告