Tensorboard 标量绘图,横轴为纪元号

Tensorboard scalar plotting with epoch number on the horizontal axis

我是 TensorFlow 的新手,最近开始尝试使用 Tensorboard 进行数据可视化。 我想知道是否可以转换监控标量的水平轴(我监控训练和验证的准确性和损失)以显示纪元数而不是迭代数。 我能想到的唯一方法是将采样频率更改为每个时期一个,但我有兴趣保持原始采样分辨率。

有没有更好的方法?

是的,你可以通过将纪元号传递给add_summary()方法的global_step参数来做到这一点:

summary_writer = tf.summary.FileWriter(log_dir)

my_summary = session.run(my_summary_op, feed_dict)
summary_writer.add_summary(my_summary, global_step=epoch_number)