Skip to content

Expose disabling decoder loading for classification and regression tasks #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Sllambias merged 1 commit into from
May 14, 2026
Merged

Expose disabling decoder loading for classification and regression tasks #3

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions asparagus/modules/lightning_modules/clsreg_module.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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__(
Expand All @@ -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
Expand Down
1 change: 1 addition & 0 deletions asparagus/pipeline/run/finetune_cls.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
1 change: 1 addition & 0 deletions asparagus/pipeline/run/finetune_reg.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
3 changes: 2 additions & 1 deletion configs/default_finetune_cls.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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
3 changes: 2 additions & 1 deletion configs/default_finetune_reg.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Loading