diff --git a/src/batdetect2/train/checkpoints.py b/src/batdetect2/train/checkpoints.py index ef69ca6..fa90973 100644 --- a/src/batdetect2/train/checkpoints.py +++ b/src/batdetect2/train/checkpoints.py @@ -34,6 +34,8 @@ def build_checkpoint_callback( if checkpoint_dir is None: checkpoint_dir = config.checkpoint_dir + checkpoint_dir = Path(checkpoint_dir) + if experiment_name is not None: checkpoint_dir = checkpoint_dir / experiment_name