new flag to resume wandb run

useful after crash or when splitting training into multiple jobs
mllam · May 1, 2024 · 72272bc · 72272bc
1 parent 96eea78
commit 72272bc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train_model.py b/train_model.py
@@ -74,6 +74,9 @@ def main():
         type=str,
         help="Path to load model parameters from (default: None)",
     )
+    parser.add_argument(
+        "--resume_run", type=str, help="Run ID to resume (default: None)"
+    )
     parser.add_argument(
         "--restore_opt",
         type=int,