复现

[youcandoitooo]

为什么我在训练的时候显存会一直增加到爆了

[xiaoze0801]

我训练3D数据的时候也是一直增加到运行不了

[lovelyasfds]

具体的代码我忘记了在哪，造成显存泄露的原因：loss_sum+=loss改成了loss_sum+=loss.item即可。