会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
嶙羽
博客园
首页
新随笔
联系
订阅
管理
2020年6月30日
Pytorch加载预训练模型的坑
摘要: 保存模型: def save(model, model_path): torch.save(model.state_dict(), model_path) 加载模型: def load(model, model_path): model.load_state_dict(torch.load(mode
阅读全文
posted @ 2020-06-30 13:25 嶙羽
阅读(2277)
评论(0)
推荐(0)
编辑
公告