updated logger to include global_step

jshuadvd · Jul 9, 2024 · 947bd63 · 947bd63
1 parent 64054e5
commit 947bd63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -263,6 +263,7 @@ def train(
         wandb.log(
             {
                 "epoch": epoch,
+                "global_step": global_step,
                 "train_loss": avg_train_loss,
                 "train_perplexity": train_perplexity,
                 "val_loss": avg_val_loss,