我正在训练一个动态神经网络,这意味着我会在每个时期调整架构并获得不同的计算图。 我想使用 tensorboard 为每个时期绘制图形,但是当我在每个时期结束时使用 SummaryWriter.add_graph() 时,它只会覆盖前一个时期。
关于如何使用 pytorch + tensorboard 绘制多张图有什么想法吗?这似乎是可以实现的,因为每个图表都有一个“标签”,但我找不到更改此标签以绘制其中多个图表的选项。
谢谢, 埃拉德
最佳答案
如果您仍想使用 SummaryWriter,可以选择使用方法“add_scalars”:
例子:
summary.add_scalars(f'loss/check_info', {
'score': score[iteration],
'score_nf': score_nf[iteration],
}, iteration)
https://stackoverflow.com/questions/60276194/