摘要: 参考:https://www.jianshu.com/p/aee6a3d72014 再加载预训练模型时,最好指定map_location。因为如果程序之前在GPU上运行,那么模型就会被存为torch.cuda.Tensor,这样加载时会默认将数据加载至显存。如果运行该程序的计算机中没有GPU,加载就 阅读全文
posted @ 2019-04-30 14:23 慢行厚积 阅读(1911) 评论(0) 推荐(0) 编辑