Skip to content

Commit 3f88187

Browse files
authored
Update checkpoint_utils.py to match new web urls in pretrained_models.py
1 parent e0ccacf commit 3f88187

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/super_gradients/training/utils/checkpoint_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1589,7 +1589,7 @@ def load_pretrained_weights(model: torch.nn.Module, architecture: str, pretraine
15891589
if url.startswith("file://") or os.path.exists(url):
15901590
pretrained_state_dict = torch.load(url.replace("file://", ""), map_location="cpu")
15911591
else:
1592-
unique_filename = url.split("https://sghub.deci.ai/models/")[1].replace("/", "_").replace(" ", "_")
1592+
unique_filename = url.split("https://sg-hub-nv.s3.amazonaws.com/models/")[1].replace("/", "_").replace(" ", "_")
15931593
map_location = torch.device("cpu")
15941594
with wait_for_the_master(get_local_rank()):
15951595
pretrained_state_dict = load_state_dict_from_url(url=url, map_location=map_location, file_name=unique_filename)

0 commit comments

Comments
 (0)