バッチごとに一部のメトリクスをログに記録し、他のメトリクスはエポックごとにのみ記録したい場合はどうすればよいですか？ - Weights & Biases Documentation

各バッチで特定のメトリクスをログに記録し、チャートを標準化するには、メトリクスと一緒に任意の x 軸の値をログに記録します。カスタムプロットでは、「編集」をクリックしてカスタムの x 軸を選択します。

import wandb

with wandb.init() as run:
    # batch_idx を x 軸の値として、loss をログに記録
    run.log({"batch": batch_idx, "loss": 0.3})
    # epoch を x 軸の値として、val_acc をログに記録
    run.log({"epoch": epoch, "val_acc": 0.94})

⌘I