pytorch训练过程中释放显存

在python程序中需要显存暴增的问题，可以试试
在代码中插入

torch.cuda.empty_cache()