tuntunmmd的博客

问题：torch.backends.cudnn.benchmark = True在利用pytorch进行网络训练时总会见到这行代码，这句有什么用呢？查了一些资料才知道是个非必须项，那加了有什么用呢？有没有什么准则呢？用途：大部分情况下，设置这个 flag 可以让内置的 cuDNN 的 auto-tuner 自动寻找最适合当前配置的高效算法，来达到优化运行效率的问题。准则：...

lossVal = 0lossValNorm = 0losstensorMean = 0 for i, (input, target) in enumerate (dataLoader): target = target.cuda(async=True) varInput = torch.autograd.Variable(input, volatile=Tr...

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版本环境下跑，会报出如上错误，应作如下修改原...