Disable wandb by default

ChenFengYe · Jan 9, 2023 · 6643f17 · 6643f17
1 parent 93dcc23
commit 6643f17
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 8 deletions.
diff --git a/configs/config_mld_humanml3d.yaml b/configs/config_mld_humanml3d.yaml
@@ -77,6 +77,6 @@ LOGGER:
   VAL_EVERY_STEPS: 200
   TENSORBOARD: True
   WANDB:
-    PROJECT: None
+    PROJECT: null
     OFFLINE: False
-    RESUME_ID: None
+    RESUME_ID: null
diff --git a/configs/config_novae_humanml3d.yaml b/configs/config_novae_humanml3d.yaml
@@ -79,6 +79,6 @@ LOGGER:
   VAL_EVERY_STEPS: 200
   TENSORBOARD: True
   WANDB:
-    PROJECT: None
+    PROJECT: null
     OFFLINE: False
-    RESUME_ID: None
+    RESUME_ID: null
diff --git a/configs/config_vae_humanml3d.yaml b/configs/config_vae_humanml3d.yaml
@@ -77,6 +77,6 @@ LOGGER:
   VAL_EVERY_STEPS: 200
   TENSORBOARD: True
   WANDB:
-    PROJECT: None
+    PROJECT: null
     OFFLINE: False
-    RESUME_ID: None
+    RESUME_ID: null
diff --git a/train.py b/train.py
@@ -61,7 +61,7 @@ def main():
 
     # tensorboard logger and wandb logger
     loggers = []
-    if not isinstance(cfg.LOGGER.WANDB.PROJECT, str):
+    if cfg.LOGGER.WANDB.PROJECT:
         wandb_logger = pl_loggers.WandbLogger(
             project=cfg.LOGGER.WANDB.PROJECT,
             offline=cfg.LOGGER.WANDB.OFFLINE,
@@ -74,7 +74,6 @@ def main():
         )
         loggers.append(wandb_logger)
     if cfg.LOGGER.TENSORBOARD:
-        print(cfg.LOGGER.WANDB.PROJECT)
         tb_logger = pl_loggers.TensorBoardLogger(save_dir=cfg.FOLDER_EXP,
                                                  sub_dir="tensorboard",
                                                  version="",