代码详解：用Pytorch训练快速神经网络的9个技巧

发布时间：2019-08-19 11:16:17 所属栏目：建站来源：读芯术

导读：副标题#e# 事实上，你的模型可能还停留在石器时代的水平。估计你还在用32位精度或*GASP(一般活动仿真语言)*训练，甚至可能只在单GPU上训练。如果市面上有99个加速指南，但你可能只看过1个?(没错，就是这样)。但这份终极指南，会一步步教你清除模型中所有的(

刚开始你可能会觉得压力很大，但其实只需做两件事: 1)将你的模型移动到GPU上，2)在用其运行数据时，把数据导至GPU中。

# put model on GPU 
model.cuda(0) 
 
# put data on gpu (cuda on a variable returns a cuda copy) 
xx = x.cuda(0) 
 
# runs on GPU now 
model(x)

如果使用Lightning，则不需要对代码做任何操作。只需设置标记

（编辑：西安站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!