会员
周边
众包
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
amarr
博客园
首页
新随笔
联系
订阅
管理
2019年3月18日
pytorch保存和加载cpu,GPU,以及多GPU模型
摘要: https://www.jianshu.com/p/4905bf8e06e5 上面这个链接主要给出了PyTorch如何保存和加载模型 今天遇到了单GPU保存模型,然后多GPU加载模型出现错误的情况。在此记录。 由于多GPU的模型参数会多出‘module.’这个前缀,所以有时要加上有时要去掉。
阅读全文
posted @ 2019-03-18 17:09 Emma00
阅读(1344)
评论(0)
推荐(0)
编辑
公告