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

import wandb

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