Allow setting wandb project names

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,

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,

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,

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,
     )

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,

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,

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,

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,

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

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

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

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

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

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
+  log_to_stdout: True

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
+  log_to_stdout: True

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