Change default weights filename + default branch is main

cc @sgugger
pull/440/head
Julien Chaumond 4 years ago
parent 8c270ed7bf
commit dd12f9f2a5

@ -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()
)

@ -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(

Loading…
Cancel
Save