From 3f6c8f53a7aa7eac8138b2c918997826acd8cddc Mon Sep 17 00:00:00 2001 From: shaydeci <80472096+shaydeci@users.noreply.github.com> Date: Sun, 12 Jun 2022 10:20:25 +0300 Subject: [PATCH] bug fixed (#215) --- src/super_gradients/training/utils/checkpoint_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/super_gradients/training/utils/checkpoint_utils.py b/src/super_gradients/training/utils/checkpoint_utils.py index 97a02b811f..fcc9691f9a 100644 --- 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':