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

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

例如

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

如何挽救这20个损失？

最佳答案

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

import torch

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


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

关于python - 如何在附加模式下保存pytorch张量，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/54570525/