clean up model_tag handling across scripts a bit more.

2025-12-27 20:01:09 -08:00
parent 2874eda59a ea4229851b
commit 088726aa7d
4 changed files with 11 additions and 7 deletions
@@ -207,7 +207,7 @@ while True:

    # save checkpoint at the end of the run (only on master process)
    if master_process and last_step and not dry_run:
-        output_dirname = f"d{depth}" # e.g. d12
+        output_dirname = model_tag if model_tag else f"d{depth}" # e.g. d12
        checkpoint_dir = os.path.join(base_dir, "mid_checkpoints", output_dirname)
        save_checkpoint(
            checkpoint_dir,