Skip to content

Commit 6643f17

Browse files
committed
Disable wandb by default
1 parent 93dcc23 commit 6643f17

4 files changed

+7
-8
lines changed

configs/config_mld_humanml3d.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,6 @@ LOGGER:
7777
VAL_EVERY_STEPS: 200
7878
TENSORBOARD: True
7979
WANDB:
80-
PROJECT: None
80+
PROJECT: null
8181
OFFLINE: False
82-
RESUME_ID: None
82+
RESUME_ID: null

configs/config_novae_humanml3d.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,6 @@ LOGGER:
7979
VAL_EVERY_STEPS: 200
8080
TENSORBOARD: True
8181
WANDB:
82-
PROJECT: None
82+
PROJECT: null
8383
OFFLINE: False
84-
RESUME_ID: None
84+
RESUME_ID: null

configs/config_vae_humanml3d.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,6 @@ LOGGER:
7777
VAL_EVERY_STEPS: 200
7878
TENSORBOARD: True
7979
WANDB:
80-
PROJECT: None
80+
PROJECT: null
8181
OFFLINE: False
82-
RESUME_ID: None
82+
RESUME_ID: null

train.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def main():
6161

6262
# tensorboard logger and wandb logger
6363
loggers = []
64-
if not isinstance(cfg.LOGGER.WANDB.PROJECT, str):
64+
if cfg.LOGGER.WANDB.PROJECT:
6565
wandb_logger = pl_loggers.WandbLogger(
6666
project=cfg.LOGGER.WANDB.PROJECT,
6767
offline=cfg.LOGGER.WANDB.OFFLINE,
@@ -74,7 +74,6 @@ def main():
7474
)
7575
loggers.append(wandb_logger)
7676
if cfg.LOGGER.TENSORBOARD:
77-
print(cfg.LOGGER.WANDB.PROJECT)
7877
tb_logger = pl_loggers.TensorBoardLogger(save_dir=cfg.FOLDER_EXP,
7978
sub_dir="tensorboard",
8079
version="",

0 commit comments

Comments
 (0)