首页 > 解决方案 > 有没有办法在wandb中缩放图轴?

问题描述

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

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

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

标签: wandb

解决方案


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

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

推荐阅读