训练内存爆炸问题

[xianming-gu]

作者您好，我在GPU上训练时内存会不断增大导致进程被kill，我是利用network_mcmrsr里面的模型，然后在自己的训练函数和数据集上进行，每次一个epoch跑不完内存就爆了，请问有遇到过这种问题吗