From b996c1a0f5068e7f5dfe69429e59e873536754c9 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 23 Nov 2023 21:48:14 -0800 Subject: [PATCH] A few more missed hf hub entries --- timm/models/efficientnet.py | 12 ++++++------ timm/models/resnet.py | 1 + 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 3b1d5f51..6e61d1bf 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1205,27 +1205,27 @@ default_cfgs = generate_default_cfgs({ 'tf_efficientnet_b0.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0-0af12548.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 224, 224)), 'tf_efficientnet_b1.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b1-5c1377c4.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 240, 240), pool_size=(8, 8), crop_pct=0.882), 'tf_efficientnet_b2.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b2-e393ef04.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 260, 260), pool_size=(9, 9), crop_pct=0.890), 'tf_efficientnet_b3.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b3-e3bd6955.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904), 'tf_efficientnet_b4.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b4-74ee3bed.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 380, 380), pool_size=(12, 12), crop_pct=0.922), 'tf_efficientnet_b5.in1k': _cfg( url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b5-c6949ce9.pth', - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 456, 456), pool_size=(15, 15), crop_pct=0.934), diff --git a/timm/models/resnet.py b/timm/models/resnet.py index dd271c59..69e28946 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -731,6 +731,7 @@ default_cfgs = generate_default_cfgs({ url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50d_a1_0-e20cff14.pth', first_conv='conv1.0'), 'resnet50d.a2_in1k': _rcfg( + hf_hub_id='timm/', url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50d_a2_0-a3adc64d.pth', first_conv='conv1.0'), 'resnet50d.a3_in1k': _r3cfg(