如何在附加模式下保存pytorch张量

Question

如何使用torch.save()保存多个张量附加？

例如

for i in range(20):
       ......
    loss = criterion(scores, labels) 
    torch.save(loss,'loss.pt')

如何挽救这20条损失？

Answer 1

可能无法直接附加到文件，至少，我找不到与此相关的文档。但是，在您的示例中，更好的方法是将附加到列表中，然后保存在末尾。

import torch

losses = []
for i in range(20):
    #   ......
    loss = criterion(scores, labels) 
    losses.append(loss.item())


torch.save(losses, 'loss.pt')