pytorch - 在 Pytorch tensorboard 中绘制几张图

Question

我正在训练一个动态神经网络，这意味着我会在每个时期调整架构并获得不同的计算图。我想使用 tensorboard 为每个时期绘制图形，但是当我在每个时期结束时使用 SummaryWriter.add_graph() 时，它只会覆盖前一个时期。

关于如何使用 pytorch + tensorboard 绘制多张图有什么想法吗？这似乎是可以实现的，因为每个图表都有一个“标签”，但我找不到更改此标签以绘制其中多个图表的选项。

谢谢，埃拉德

Answer 1

如果您仍想使用 SummaryWriter，可以选择使用方法“add_scalars”:

例子:

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