Correct WandB log step

2023-08-25 17:12:29 +08:00
parent d7cd98633d
commit 6def77d4ba
1 changed files with 1 additions and 1 deletions
@@ -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}")