pytorch在训练时，显示内存不够 cuda out of memory

在跑代码时，训练集跑的时候正常，在测试验证集和保存模型的时候，提示内存不够，直接报错终止，这种一般并不是batchsize过大导致的，不然训练集也运行不起来，最简单直接的方法，就是把验证集放到cpu上，一般验证集的数据比较小，也很快就完成了
例如：

imgn_val =Variable(imgn_val.cuda(), requires_grad=False)
out_val = torch.clamp(model(imgn_val), 0., 1.)

这里把验证集转到cpu里，去掉 .cuda()即可