dry_run fix

run_cspca.py

@@ -149,9 +149,6 @@ def parse_args():
         type=int,
         help="run validation after this number of epochs, default 1 to run every epoch",
     )
-    parser.add_argument(
-        "--dry_run", action="store_true", help="Run the script in dry-run mode (default: False)"
-    )
     parser.add_argument("--checkpoint_cspca", default=None, help="load existing checkpoint")
     parser.add_argument(
         "--num_seeds", default=20, type=int, help="number of seeds to be run to build CI"
@@ -200,14 +197,4 @@ if __name__ == "__main__":
     if args.device == torch.device("cuda"):
         torch.backends.cudnn.benchmark = True
 
-    if args.dry_run:
-        logging.info("Dry run mode enabled.")
-        args.epochs = 1
-        args.batch_size = 2
-        args.workers = 0
-        args.num_seeds = 1
-        args.wandb = False
-        args.tile_size = 10
-        args.tile_count = 5
-
     main_worker(args)

run_pirads.py

@@ -247,9 +247,6 @@ def parse_args():
         type=int,
         help="run validation after this number of epochs, default 1 to run every epoch",
     )
-    parser.add_argument(
-        "--dry_run", action="store_true", help="Run the script in dry-run mode (default: False)"
-    )
     args = parser.parse_args()
     if args.config:
         with open(args.config) as config_file:
@@ -289,16 +286,6 @@ if __name__ == "__main__":
         logging.error("Model checkpoint path not provided. Quitting.")
         sys.exit(1)
 
-    if args.dry_run:
-        logging.info("Dry run mode enabled.")
-        args.epochs = 1
-        args.batch_size = 2
-        args.workers = 0
-        args.num_seeds = 1
-        args.wandb = False
-        args.tile_size = 10
-        args.tile_count = 5
-
     mode_wandb = "online" if args.wandb and args.mode != "test" else "disabled"
 
     config_wandb = {