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

各バッチで特定のメトリクスをログし、プロットを統一するには、必要な x-axis の値もメトリクスとあわせてログします。カスタムプロットで Edit をクリックし、カスタムの x-axis を選択します。

import wandb

with wandb.init() as run:
    run.log({"batch": batch_idx, "loss": 0.3})
    run.log({"epoch": epoch, "val_acc": 0.94})

⌘I