会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Parallax
初极狭,才通人;复行数十步,豁然开朗。
博客园
首页
新随笔
联系
订阅
管理
2021年1月28日
pytorch模型与参数的GPU,cpu加载转换
摘要: 我们在使用pytorch的过程,经常会需要加载模型参数,不管是别人提供给我们的模型参数,还是我们自己训练的模型参数,那么加载模型参数就会碰到一些情况,即GPU模型和CPU模型,这两种模型是不能混为一谈的,下面分情况进行操作说明。 情况一:模型是GPU模型,预加载的训练参数也是GPU;模型是CPU模型
阅读全文
posted @ 2021-01-28 19:50 Parallax
阅读(672)
评论(0)
推荐(0)
编辑
公告