- 博客(3)
- 资源 (2)
- 收藏
- 关注
RSS订阅转载 pytroch中指定GPU或者CPU写法
之前习惯用.cpu()，.cuda()来指定．现在不要显示的指定是gpu, cpu之类的. 利用.to()来执行# at beginning of the scriptdevice = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")...# then whenever you get a new Tenso...
2018-11-28 20:43:02
6673
原创 pytorch0.4使用注意
1.梯度１．Variable()中，requires_grad=Fasle时不需要更新梯度， 适用于冻结某些层的梯度；２．volatile=True相当于requires_grad=False，适用于测试阶段，不需要反向传播。在torch>=0.4中，这个现在已经取消了，使用with torch.no_grad()或者torch.set_grad_enable(grad_mode)来替代...
2018-11-28 20:38:55
3766
转载 Ubuntu建立新用户教程
博客链接：https://blog.csdn.net/longgeaisisi/article/details/78879119
2018-11-24 19:45:39
2131
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人