在Pytorch张量板上绘制几个图

Question

我正在训练动态神经网络，这意味着我每个时期都会调整架构并获得不同的计算图。我想使用张量板绘制每个时期的图形，但是当我在每个时期的末尾使用SummaryWriter.add_graph（）时，它只会覆盖前一个时期。

任何想法如何使用pytorch +张量板绘制多个图形？每个图都有一个“标签”似乎可以实现，但我发现没有选择更改此标签以绘制其中几个标签的选项。

谢谢，埃拉德

Answer 1

如果仍要使用SummaryWriter，则可以选择使用方法“ add_scalars”：

示例：

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