![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
萌新调包员
这个作者很懒,什么都没留下…
展开
-
pytorch——torch.backends.cudnn.benchmark = True
问题:torch.backends.cudnn.benchmark = True在利用pytorch进行网络训练时总会见到这行代码,这句有什么用呢?查了一些资料才知道是个非必须项,那加了有什么用呢?有没有什么准则呢?用途:大部分情况下,设置这个 flag 可以让内置的 cuDNN 的 auto-tuner 自动寻找最适合当前配置的高效算法,来达到优化运行效率的问题。准则:...原创 2019-05-15 09:47:02 · 3389 阅读 · 0 评论 -
一个内存溢出Bug——RuntimeError: CUDA out of memory.
lossVal = 0lossValNorm = 0losstensorMean = 0 for i, (input, target) in enumerate (dataLoader): target = target.cuda(async=True) varInput = torch.autograd.Variable(input, volatile=Tr...原创 2019-05-20 14:11:05 · 7490 阅读 · 7 评论 -
一个torch版本报错
lossVal += losstensor.data[0]IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number解释:如果是torch0.3.x版本的代码在torch0.4.x版本环境下跑,会报出如上错误,应作如下修改原...原创 2019-05-11 09:35:58 · 359 阅读 · 0 评论