会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
慢行厚积
博客园
首页
新随笔
联系
订阅
管理
2019年4月30日
pytorch加载数据的方法-没弄,打算弄
摘要: 参考:https://www.jianshu.com/p/aee6a3d72014 再加载预训练模型时,最好指定map_location。因为如果程序之前在GPU上运行,那么模型就会被存为torch.cuda.Tensor,这样加载时会默认将数据加载至显存。如果运行该程序的计算机中没有GPU,加载就
阅读全文
posted @ 2019-04-30 14:23 慢行厚积
阅读(1911)
评论(0)
推荐(0)
编辑
公告