diff --git a/scripts/train.py b/scripts/train.py index b4d89be2d..8d7d85a2a 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -107,7 +107,7 @@ def main(cfg: TrainConfig) -> None: wandb_dir = Path(cfg.save_folder) / "wandb" wandb_dir.mkdir(parents=True, exist_ok=True) wandb.init( - dir=wandb_dir, + dir=str(wandb_dir), project=cfg.wandb.project, entity=cfg.wandb.entity, group=cfg.wandb.group,