From 947bd633932ffee9a1c657169446fc100b23efc7 Mon Sep 17 00:00:00 2001
From: Joshua David <jshuadvd@icloud.com>
Date: Mon, 8 Jul 2024 23:01:43 -0700
Subject: [PATCH] updated logger to include global_step

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index bda7d72..e6c1f88 100644
--- 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,