Pytorch 训练与测试时爆显存(out of memory)的一个解决方案