wandb - 有没有办法在wandb中缩放图轴？

Question

我正在处理重量和偏差（wandb）。
但是，它会逐步记录。在比较运行时，这会使情节令人不安。
例如，我有一个运行 A 和运行 B（假设它们使用相同的数据集运行）。
运行 A: 30epochs, 4 batch, 200step/epoch
运行 B: 30epochs, 8 batch, 100step/epoch

然后，当与运行 B 一起显示时，运行 A 的图在 x 轴上变长（在本例中为双倍）。

如何缩放 x 轴取决于训练后的运行？

Answer 1

您可以通过单击铅笔图标然后选择不同的 x 轴来更改通过图表设置使用的 x 轴。例如，在您的情况下，您可以选择“epoch”而不是“steps”。只需确保将“纪元”记录到您的图表中，例如：

steps_per_epoch = n_samples / batch_size
epoch = current_step / steps_per_epoch
wandb.log({"epoch":epoch, ...})