есть ли способ масштабировать ось графиков в wandb?

Question

Я работаю с весом и уклоном (wandb).
Тем не менее, он записывается шаг за шагом. И это делает сюжет тревожным при сравнении прогонов.
Например, у меня есть запуск A и запуск B(предположим, что они работают с одним и тем же набором данных).
запуск A: 30 этапов, 4 пакета, 200 этапов / эпоха,
запуск B: 30 этапов, 8 пакетов, 100 этапов / эпоха

тогда график прогона A становится длиннее (в данном случае вдвое) по оси x, когда он отображается с прогоном B.

Как масштабировать ось x в зависимости от пробежек ПОСЛЕ тренировки?

Answer 1

Вы можете изменить используемую ось x в настройках диаграммы, щелкнув значок карандаша и выбрав другую ось x. Например, в вашем случае вы можете выбрать «эпоху» вместо «шаги». Просто убедитесь, что в ваших графиках записана "эпоха", например:

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