搜索内容

最新文章
  • PyTorch保存Tensor日志会持续占用显存
    Python

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

    已阅: 0 用列表存放在GPU的Tensor,会一直占用显存,如果只是为了记录日志,可以用loss.item()例如,losses_logs.append([loss1, loss2])会持续占用显存。 而losses_logs.append([loss1.item(), loss2.item()])可以保留数值,但不会占用显存。
    MWHLS 14小时前