diff --git a/timm/models/helpers.py b/timm/models/helpers.py index fe1f509d..0908694d 100644 --- a/timm/models/helpers.py +++ b/timm/models/helpers.py @@ -198,7 +198,7 @@ def load_pretrained(model, cfg=None, num_classes=1000, in_chans=3, filter_fn=Non if hf_checkpoint is not None: # TODO, progress is ignored. - url = hf_hub_url(hf_checkpoint, "pytorch_model.pth", revision=hf_revision) + url = hf_hub_url(hf_checkpoint, "pytorch_model.bin", revision=hf_revision) cached_filed = cached_download( url, library_name="timm", library_version=__version__, cache_dir=get_cache_dir() ) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 64760a6d..3005df66 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -53,7 +53,7 @@ default_cfgs = { interpolation='bicubic'), 'resnet50d': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnet50d_ra2-464e36ba.pth', - hf_revision="master", hf_model_id="sgugger/resnet50d", + hf_revision="main", hf_model_id="sgugger/resnet50d", interpolation='bicubic', first_conv='conv1.0'), 'resnet101': _cfg(url='', interpolation='bicubic'), 'resnet101d': _cfg(