PyTorch保存Tensor日志会持续占用显存

MWHLS • • Python, Pytorch, 计算机语言
已阅： 1
  • 用列表存放在GPU的Tensor，会一直占用显存，如果只是为了记录日志，可以用loss.item()
    • 例如，losses_logs.append([loss1, loss2])会持续占用显存。
    • 而losses_logs.append([loss1.item(), loss2.item()])可以保留数值，但不会占用显存。

版权声明：
作者：MWHLS
链接：https://mwhls.top/5059.html
来源：无镣之涯
文章版权归作者所有，未经允许请勿转载。

THE END
分享
二维码
打赏
< <上一篇
下一篇>>