基于pytorch的python程序

小白最近在跑一个关于GAN的代码，但在这个代码中调用的一个程序中出现了一些问题。总的表现为超出显存。但仔细排查后发现程序每次循环的时候都把它自己定义的模型给送到GPU 了，导致显存激增，且循环完毕也不释放。想求教各路大神怎么解决这个问题。改网络结构的话，以我目前的水平是改不了。希望大神们可以帮忙！