From 72272bc5362bdfbcac6facf9fd08979e495ad5bd Mon Sep 17 00:00:00 2001
From: Simon Adamov <simon.adamov@mailbox.org>
Date: Wed, 1 May 2024 22:02:17 +0200
Subject: [PATCH] new flag to resume wandb run useful after crash or when
 splitting training into multiple jobs

---
 train_model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/train_model.py b/train_model.py
index bfadbb34..2c579050 100644
--- 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,