Skip to content

Commit b996c1a

Browse files
committed
A few more missed hf hub entries
1 parent f9c9f6b commit b996c1a

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

timm/models/efficientnet.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1205,27 +1205,27 @@ def _cfg(url='', **kwargs):
12051205

12061206
'tf_efficientnet_b0.in1k': _cfg(
12071207
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0-0af12548.pth',
1208-
#hf_hub_id='timm/',
1208+
hf_hub_id='timm/',
12091209
input_size=(3, 224, 224)),
12101210
'tf_efficientnet_b1.in1k': _cfg(
12111211
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b1-5c1377c4.pth',
1212-
#hf_hub_id='timm/',
1212+
hf_hub_id='timm/',
12131213
input_size=(3, 240, 240), pool_size=(8, 8), crop_pct=0.882),
12141214
'tf_efficientnet_b2.in1k': _cfg(
12151215
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b2-e393ef04.pth',
1216-
#hf_hub_id='timm/',
1216+
hf_hub_id='timm/',
12171217
input_size=(3, 260, 260), pool_size=(9, 9), crop_pct=0.890),
12181218
'tf_efficientnet_b3.in1k': _cfg(
12191219
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b3-e3bd6955.pth',
1220-
#hf_hub_id='timm/',
1220+
hf_hub_id='timm/',
12211221
input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904),
12221222
'tf_efficientnet_b4.in1k': _cfg(
12231223
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b4-74ee3bed.pth',
1224-
#hf_hub_id='timm/',
1224+
hf_hub_id='timm/',
12251225
input_size=(3, 380, 380), pool_size=(12, 12), crop_pct=0.922),
12261226
'tf_efficientnet_b5.in1k': _cfg(
12271227
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b5-c6949ce9.pth',
1228-
#hf_hub_id='timm/',
1228+
hf_hub_id='timm/',
12291229
input_size=(3, 456, 456), pool_size=(15, 15), crop_pct=0.934),
12301230

12311231

timm/models/resnet.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -731,6 +731,7 @@ def _gcfg(url='', **kwargs):
731731
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50d_a1_0-e20cff14.pth',
732732
first_conv='conv1.0'),
733733
'resnet50d.a2_in1k': _rcfg(
734+
hf_hub_id='timm/',
734735
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50d_a2_0-a3adc64d.pth',
735736
first_conv='conv1.0'),
736737
'resnet50d.a3_in1k': _r3cfg(

0 commit comments

Comments
 (0)