From 02e6bfc75e740cf63dff75546d79434bef808a21 Mon Sep 17 00:00:00 2001 From: lukasugar Date: Fri, 15 May 2026 17:53:47 +0200 Subject: [PATCH] Allow setting wandb project names --- asparagus/pipeline/run/finetune_cls.py | 2 +- asparagus/pipeline/run/finetune_reg.py | 2 +- asparagus/pipeline/run/finetune_seg.py | 2 +- asparagus/pipeline/run/linear_probe.py | 2 +- asparagus/pipeline/run/pretrain.py | 2 +- asparagus/pipeline/run/train_cls.py | 1 + asparagus/pipeline/run/train_reg.py | 1 + asparagus/pipeline/run/train_seg.py | 2 +- configs/default_finetune_cls.yaml | 1 + configs/default_finetune_reg.yaml | 1 + configs/default_finetune_seg.yaml | 1 + configs/default_linear_probe.yaml | 1 + configs/default_pretrain.yaml | 1 + configs/default_train_cls.yaml | 3 ++- configs/default_train_reg.yaml | 3 ++- configs/default_train_seg.yaml | 1 + 16 files changed, 18 insertions(+), 8 deletions(-) diff --git a/asparagus/pipeline/run/finetune_cls.py b/asparagus/pipeline/run/finetune_cls.py index 0e9bd0e..3e7ef18 100644 --- a/asparagus/pipeline/run/finetune_cls.py +++ b/asparagus/pipeline/run/finetune_cls.py @@ -51,7 +51,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, - wandb_project="Finetune", + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/finetune_reg.py b/asparagus/pipeline/run/finetune_reg.py index 8c37f4c..976d292 100644 --- a/asparagus/pipeline/run/finetune_reg.py +++ b/asparagus/pipeline/run/finetune_reg.py @@ -51,7 +51,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_experiment=HydraConfig.get().job.config_name, wandb_entity=cfg.logger.wandb_entity, - wandb_project="Finetune", + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/finetune_seg.py b/asparagus/pipeline/run/finetune_seg.py index 70899a9..acfc16a 100644 --- a/asparagus/pipeline/run/finetune_seg.py +++ b/asparagus/pipeline/run/finetune_seg.py @@ -53,7 +53,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, - wandb_project=cfg.task, + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/linear_probe.py b/asparagus/pipeline/run/linear_probe.py index 33a0fcc..0bc2566 100644 --- a/asparagus/pipeline/run/linear_probe.py +++ b/asparagus/pipeline/run/linear_probe.py @@ -49,7 +49,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, - wandb_project="LinearProbe", + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, ) diff --git a/asparagus/pipeline/run/pretrain.py b/asparagus/pipeline/run/pretrain.py index b323275..76a66f2 100644 --- a/asparagus/pipeline/run/pretrain.py +++ b/asparagus/pipeline/run/pretrain.py @@ -44,7 +44,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, - wandb_project="Pretrain", + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/train_cls.py b/asparagus/pipeline/run/train_cls.py index dacfcba..9526a14 100644 --- a/asparagus/pipeline/run/train_cls.py +++ b/asparagus/pipeline/run/train_cls.py @@ -50,6 +50,7 @@ def main(cfg: DictConfig) -> None: wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, wandb_entity=cfg.logger.wandb_entity, + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/train_reg.py b/asparagus/pipeline/run/train_reg.py index 781f975..9a469b6 100644 --- a/asparagus/pipeline/run/train_reg.py +++ b/asparagus/pipeline/run/train_reg.py @@ -50,6 +50,7 @@ def main(cfg: DictConfig) -> None: wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, wandb_entity=cfg.logger.wandb_entity, + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/asparagus/pipeline/run/train_seg.py b/asparagus/pipeline/run/train_seg.py index bdeaacc..5403ca2 100644 --- a/asparagus/pipeline/run/train_seg.py +++ b/asparagus/pipeline/run/train_seg.py @@ -49,7 +49,7 @@ def main(cfg: DictConfig) -> None: version=version_store.version, wandb_config=logging_safe_cfg, wandb_experiment=HydraConfig.get().job.config_name, - wandb_project=cfg.task, + wandb_project=cfg.logger.wandb_project, wandb_logging=cfg.logger.wandb_logging, mlflow_logging=cfg.logger.mlflow_logging, log_to_stdout=cfg.logger.log_to_stdout, diff --git a/configs/default_finetune_cls.yaml b/configs/default_finetune_cls.yaml index 8d2b0ec..e0e2102 100644 --- a/configs/default_finetune_cls.yaml +++ b/configs/default_finetune_cls.yaml @@ -51,6 +51,7 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: Finetune mlflow_logging: False log_every_n_steps: 250 log_images_every_n_epoch: 5 diff --git a/configs/default_finetune_reg.yaml b/configs/default_finetune_reg.yaml index 572d755..8722511 100644 --- a/configs/default_finetune_reg.yaml +++ b/configs/default_finetune_reg.yaml @@ -47,6 +47,7 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: Finetune mlflow_logging: False log_every_n_steps: 250 log_images_every_n_epoch: 5 diff --git a/configs/default_finetune_seg.yaml b/configs/default_finetune_seg.yaml index bee87b9..c2106af 100644 --- a/configs/default_finetune_seg.yaml +++ b/configs/default_finetune_seg.yaml @@ -51,6 +51,7 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: ${task} mlflow_logging: False log_every_n_steps: 50 log_images_every_n_epoch: 5 diff --git a/configs/default_linear_probe.yaml b/configs/default_linear_probe.yaml index 528d083..acef864 100644 --- a/configs/default_linear_probe.yaml +++ b/configs/default_linear_probe.yaml @@ -47,6 +47,7 @@ logger: progress_bar: True wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: LinearProbe mlflow_logging: False log_every_n_steps: 25 log_images_every_n_epoch: 9999999 diff --git a/configs/default_pretrain.yaml b/configs/default_pretrain.yaml index 6b6f8f1..d09c083 100644 --- a/configs/default_pretrain.yaml +++ b/configs/default_pretrain.yaml @@ -70,6 +70,7 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: Pretrain mlflow_logging: False log_every_n_steps: 250 log_images_every_n_epoch: 5 diff --git a/configs/default_train_cls.yaml b/configs/default_train_cls.yaml index 59df966..be463ca 100644 --- a/configs/default_train_cls.yaml +++ b/configs/default_train_cls.yaml @@ -49,7 +49,8 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: Asparagus mlflow_logging: False log_every_n_steps: 250 log_images_every_n_epoch: 5 - log_to_stdout: True \ No newline at end of file + log_to_stdout: True diff --git a/configs/default_train_reg.yaml b/configs/default_train_reg.yaml index d8497a9..38efec9 100644 --- a/configs/default_train_reg.yaml +++ b/configs/default_train_reg.yaml @@ -48,7 +48,8 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: Asparagus mlflow_logging: False log_every_n_steps: 250 log_images_every_n_epoch: 1 - log_to_stdout: True \ No newline at end of file + log_to_stdout: True diff --git a/configs/default_train_seg.yaml b/configs/default_train_seg.yaml index 03a8909..79c65ce 100644 --- a/configs/default_train_seg.yaml +++ b/configs/default_train_seg.yaml @@ -48,6 +48,7 @@ logger: wandb_log_model: False wandb_logging: True wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"} + wandb_project: ${task} mlflow_logging: False log_images_every_n_epoch: 1 log_every_n_steps: 250