Skip to content

Commit 656af4d

Browse files
committed
Fix wandb config
1 parent 1c7d6ce commit 656af4d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

sleap_nn/training/model_trainer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -707,7 +707,10 @@ def _setup_loggers_callbacks(self, viz_train_dataset, viz_val_dataset):
707707
if RANK in [0, -1]:
708708
wandb.login(key=self.config.trainer_config.wandb.api_key)
709709
wandb.init(
710-
dir=self.config.trainer_config.save_ckpt_path,
710+
dir=(
711+
Path(self.config.trainer_config.ckpt_dir)
712+
/ self.config.trainer_config.run_name
713+
).as_posix(),
711714
project=self.config.trainer_config.wandb.project,
712715
entity=self.config.trainer_config.wandb.entity,
713716
name=self.config.trainer_config.wandb.name,

0 commit comments

Comments
 (0)