fix: ensure checkpoint is path

This commit is contained in:
Santiago Martinez Balvanera 2026-05-04 22:56:34 +01:00
parent aa36df668f
commit c27e7f9f52

View File

@ -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