From dd12f9f2a5aeb5b63e42294b7a8975f190a19f96 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Thu, 18 Feb 2021 11:51:50 +0100 Subject: [PATCH] Change default weights filename + default branch is main cc @sgugger --- timm/models/helpers.py | 2 +- timm/models/resnet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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(