pytorch 测试时候显存溢出

cuda out of memory

解决方法：http://www.cnblogs.com/Charlene-HRI/p/10234656.html

我的问题时在测试时候需要保存每次的输出中间向量，忘记转为 cpu().numpy(),

导致显存累积，越来越大，最终溢出