improve docu of continue_from_dir

RolnickLab · Jan 6, 2024 · ca497e8 · ca497e8
1 parent 6445a5c
commit ca497e8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ocpmodels/common/flags.py b/ocpmodels/common/flags.py
@@ -87,12 +87,14 @@ def add_core_args(self):
             "--checkpoint", type=str, help="Model checkpoint to load"
         )
         self.parser.add_argument(
-            "--continue_from_dir", type=str, help="Run to continue, loading its config"
+            "--continue_from_dir",
+            type=str,
+            help="Continue an existing run, loading its config and overwriting desired arguments",
         )
         self.parser.add_argument(
             "--restart_from_dir",
             type=str,
-            help="Run to restart, loading its config and overwriting "
+            help="Restart training from an existing run, loading its config and overwriting args"
             + "from the command-line",
         )
         self.parser.add_argument(