Merge pull request #3 from lukasugar/expose_load_decoder_for_cls_and_reg

Expose disabling decoder loading for classification and regression tasks

Author: Sllambias

asparagus/modules/lightning_modules/clsreg_module.py

@@ -33,6 +33,7 @@ def __init__(
         momentum: float = 0.99,
         log_image_every_n_epochs: int = 50,
         test_output_path: str = None,
+        load_decoder: bool = True,
         repeat_stem_weights: bool = True,
     ):
         super().__init__(
@@ -50,6 +51,7 @@ def __init__(
             weight_decay=weight_decay,
             nesterov=nesterov,
             momentum=momentum,
+            load_decoder=load_decoder,
             repeat_stem_weights=repeat_stem_weights,
         )
         self.loss = None

asparagus/pipeline/run/finetune_cls.py

@@ -114,6 +114,7 @@ def main(cfg: DictConfig) -> None:
         log_image_every_n_epochs=cfg.logger.log_images_every_n_epoch,
         optimizer=cfg.model.finetune_optim,
         learning_rate=cfg.model.finetune_lr,
+        load_decoder=cfg.training.load_decoder,
         repeat_stem_weights=cfg.training.repeat_stem_weights,
         test_output_path=os.path.join(
             path_store.run_dir,

asparagus/pipeline/run/finetune_reg.py

@@ -114,6 +114,7 @@ def main(cfg: DictConfig) -> None:
         learning_rate=cfg.model.finetune_lr,
         warmup_epochs=cfg.training.warmup_epochs,
         weight_decay=cfg.model.weight_decay,
+        load_decoder=cfg.training.load_decoder,
         test_output_path=os.path.join(
             path_store.run_dir,
             "predictions",

configs/default_finetune_cls.yaml

@@ -35,6 +35,7 @@ training:
   mask_ratio: 0.0
   check_val_every_n_epoch: 3
   accumulate_grad_batches: 1
+  load_decoder: True
   repeat_stem_weights: True
 
 transforms:
@@ -53,4 +54,4 @@ logger:
   mlflow_logging: False
   log_every_n_steps: 250
   log_images_every_n_epoch: 5
-  log_to_stdout: True
+  log_to_stdout: True

configs/default_finetune_reg.yaml

@@ -32,6 +32,7 @@ training:
   accumulate_grad_batches: 1
   warmup_epochs: 0
   check_val_every_n_epoch: 1
+  load_decoder: True
 
 transforms:
   cpu_tr_transforms: CPU_clsreg_train_transforms_crop
@@ -49,4 +50,4 @@ logger:
   mlflow_logging: False
   log_every_n_steps: 250
   log_images_every_n_epoch: 5
-  log_to_stdout: True
+  log_to_stdout: True