CUDA out of memory

减小batchsize在output = model(input)前面加上 with torch.no_grad:

总结：https://segmentfault.com/a/1190000022589080