Merge pull request #353 from photomz/master

Clearer WandB log step
2023-08-25 07:55:26 -07:00
parent d7cd98633d 6def77d4ba
commit cbc2488b82
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}")