diff --git a/src/super_gradients/common/environment/checkpoints_dir_utils.py b/src/super_gradients/common/environment/checkpoints_dir_utils.py index 8cc1f7cd10..e359f56503 100644 --- a/src/super_gradients/common/environment/checkpoints_dir_utils.py +++ b/src/super_gradients/common/environment/checkpoints_dir_utils.py @@ -37,7 +37,7 @@ def get_project_checkpoints_dir_path() -> Optional[str]: checkpoints_path = os.path.join(project_root_path, "checkpoints") if not os.path.exists(checkpoints_path): - os.makedirs(checkpoints_path) + os.makedirs(checkpoints_path, exist_ok=True) logger.info(f'A checkpoints directory was just created at "{checkpoints_path}". To work with another directory, please set "ckpt_root_dir"') return checkpoints_path