会员
周边
新闻
博问
AI培训
云市场
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
DinoX的博客
博客园
首页
新随笔
联系
订阅
管理
2020年6月15日
[pytorch]模型参数保存与加载
摘要: 最简单的情况 模型保存: torch.save(model.state_dict(), PATH) 模型加载: model.load_state_dict(torch.load(PATH)) 此时保存的是一个字典,key为model中的weight或bias名,如"linear1.weight"或“
阅读全文
posted @ 2020-06-15 08:40 DinoX
阅读(639)
评论(0)
推荐(0)
编辑
公告