From 6def77d4baef39b9878643001f473e49af405c7c Mon Sep 17 00:00:00 2001
From: Markus Zhang <45726499+photomz@users.noreply.github.com>
Date: Fri, 25 Aug 2023 17:12:29 +0800
Subject: [PATCH] Correct WandB log step

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

diff --git a/train.py b/train.py
index e958538..e8321d8 100644
--- a/train.py
+++ b/train.py
@@ -271,7 +271,7 @@ while True:
                         "loss/val": losses["val"],
                         "lr": lr,
                         "mfu": running_mfu * 100,  # convert to percentage
-                    }
+                    }, step = iter_num
                 )
             except Exception as e:
                 print(f"logging to wandb failed: {e}")