Merge pull request #4 from lukasugar/expose_wand_project_name_as_param

Allow setting wandb project names

Author: Sllambias

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