Как сохранить тензор 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')