update default behavior

westny · May 7, 2024 · ac7080f · ac7080f
1 parent f361f0a
commit ac7080f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/arguments.py b/arguments.py
@@ -48,9 +48,9 @@ def str_to_bool(value: bool | str) -> bool:
 parser.add_argument('--persistent-workers', type=str_to_bool, default=True,
                     const=True, nargs="?",
                     help='if persistent workers should be used (default: True)')
-parser.add_argument('--store-model', type=str_to_bool, default=False,
+parser.add_argument('--store-model', type=str_to_bool, default=True,
                     const=True, nargs="?",
-                    help='if checkpoints should be stored (default: False)')
+                    help='if checkpoints should be stored (default: True)')
 parser.add_argument('--store-samples', type=str_to_bool, default=False,
                     const=True, nargs="?",
                     help='if samples should be stored (default: False)')

diff --git a/train.py b/train.py
@@ -52,7 +52,7 @@ def main(save_name: str) -> None:
 
     # Setup callbacks list for training
     callback_list: list[Callback] = []
-    if args.store_model:
+    if args.store_model and not args.dry_run:
         ckpt_cb = ModelCheckpoint(
             dirpath=str(ckpt_path.parent),  # Using parent directory of the checkpoint
             filename=save_name + "_{epoch:02d}",