Постройте несколько графиков в тензорной доске Pytorch.

Question

Я тренирую динамическую c нейронную сеть, что означает, что в каждую эпоху я настраиваю архитектуру и получаю различный вычислительный граф. Я хочу построить график для каждой эпохи с использованием тензорной доски, но когда я использую SummaryWriter.add_graph () в конце каждой эпохи, он просто перезаписывает предыдущую.

Любые идеи, как построить несколько графиков, используя pytorch + tensorboard? Это кажется достижимым, так как каждый график имеет «тег», но я не нашел возможности изменить этот тег, чтобы построить несколько из них.

Спасибо, Elad

Answer 1

Если вы все еще хотите использовать SummaryWriter, есть вариант использования метода "add_scalars":

Пример:

summary.add_scalars(f'loss/check_info', {
    'score': score[iteration],
    'score_nf': score_nf[iteration],
}, iteration)