bug fixed (#215)

Deci-AI · Jun 12, 2022 · 3f6c8f5 · 3f6c8f5
1 parent 425b227
commit 3f6c8f5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/super_gradients/training/utils/checkpoint_utils.py b/src/super_gradients/training/utils/checkpoint_utils.py
@@ -31,9 +31,7 @@ def get_ckpt_local_path(source_ckpt_folder_name: str, experiment_name: str, ckpt
     """
     source_ckpt_folder_name = source_ckpt_folder_name or experiment_name
     if model_checkpoints_location == 'local':
-        default_local_ckpt_path = pkg_resources.resource_filename('checkpoints',
-                                                                  source_ckpt_folder_name + os.path.sep + ckpt_name)
-        ckpt_local_path = external_checkpoint_path or default_local_ckpt_path
+        ckpt_local_path = external_checkpoint_path or pkg_resources.resource_filename('checkpoints', source_ckpt_folder_name + os.path.sep + ckpt_name)
 
     # COPY THE DATA FROM 'S3'/'URL' INTO A LOCAL DIRECTORY
     elif model_checkpoints_location.startswith('s3') or model_checkpoints_location == 'url':