Update checkpoint_utils.py to match new web urls in pretrained_models.py

src/super_gradients/recipes/arch_params/ppyoloe_l_arch_params.yaml

@@ -4,6 +4,3 @@ defaults:
 
 depth_mult: 1.0
 width_mult: 1.0
-
-backbone:
-  pretrained_weights: https://deci-pretrained-models.s3.amazonaws.com/ppyolo_e/CSPResNetb_l_pretrained.pth

src/super_gradients/recipes/arch_params/ppyoloe_m_arch_params.yaml

@@ -4,6 +4,3 @@ defaults:
 
 depth_mult: 0.67
 width_mult: 0.75
-
-backbone:
-  pretrained_weights: https://deci-pretrained-models.s3.amazonaws.com/ppyolo_e/CSPResNetb_m_pretrained.pth

src/super_gradients/recipes/arch_params/ppyoloe_s_arch_params.yaml

@@ -3,7 +3,4 @@ defaults:
   - _self_
 
 depth_mult: 0.33
-width_mult: 0.50
-
-backbone:
-  pretrained_weights: https://deci-pretrained-models.s3.amazonaws.com/ppyolo_e/CSPResNetb_s_pretrained.pth
+width_mult: 0.50

src/super_gradients/recipes/arch_params/ppyoloe_x_arch_params.yaml

@@ -4,6 +4,3 @@ defaults:
 
 depth_mult: 1.33
 width_mult: 1.25
-
-backbone:
-  pretrained_weights: https://deci-pretrained-models.s3.amazonaws.com/ppyolo_e/CSPResNetb_x_pretrained.pth

src/super_gradients/training/utils/checkpoint_utils.py

@@ -1589,7 +1589,7 @@ def load_pretrained_weights(model: torch.nn.Module, architecture: str, pretraine
     if url.startswith("file://") or os.path.exists(url):
         pretrained_state_dict = torch.load(url.replace("file://", ""), map_location="cpu")
     else:
-        unique_filename = url.split("https://sghub.deci.ai/models/")[1].replace("/", "_").replace(" ", "_")
+        unique_filename = url.split("https://sg-hub-nv.s3.amazonaws.com/models/")[1].replace("/", "_").replace(" ", "_")
         map_location = torch.device("cpu")
         with wait_for_the_master(get_local_rank()):
             pretrained_state_dict = load_state_dict_from_url(url=url, map_location=map_location, file_name=unique_filename)

tests/unit_tests/load_checkpoint_test.py

@@ -33,7 +33,7 @@ def __init__(self):
         self.assertTrue((foo.fc2.weight == bar.fc2.weight).all())
 
     def test_checkpoint_path_url(self):
-        m1 = models.get(Models.YOLO_NAS_S, num_classes=80, checkpoint_path="https://sghub.deci.ai/models/yolo_nas_s_coco.pth")
+        m1 = models.get(Models.YOLO_NAS_S, num_classes=80, checkpoint_path="https://sg-hub-nv.s3.amazonaws.com/models/yolo_nas_s_coco.pth")
         m2 = models.get(Models.YOLO_NAS_S, pretrained_weights="coco")
         m1_state = m1.state_dict()
         m2_state = m2.state_dict()

tests/unit_tests/test_predict.py

@@ -17,7 +17,6 @@ def setUp(self) -> None:
         self.images = [
             os.path.join(rootdir, "documentation", "source", "images", "examples", "countryside.jpg"),
             os.path.join(rootdir, "documentation", "source", "images", "examples", "street_busy.jpg"),
-            "https://deci-datasets-research.s3.amazonaws.com/image_samples/beatles-abbeyroad.jpg",
         ]
         self._set_images_with_targets()