Skip to content

Commit 9c11dfd

Browse files
committed
Fix fbnetv3 pretrained cfg changes
1 parent 1406cdd commit 9c11dfd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

timm/models/mobilenetv3.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,13 +77,13 @@ def _cfg(url='', **kwargs):
7777

7878
'fbnetv3_b': _cfg(
7979
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_b_224-ead5d2a1.pth',
80-
test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
80+
test_input_size=(3, 256, 256), crop_pct=0.95),
8181
'fbnetv3_d': _cfg(
8282
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_d_224-c98bce42.pth',
83-
test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
83+
test_input_size=(3, 256, 256), crop_pct=0.95),
8484
'fbnetv3_g': _cfg(
8585
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_g_240-0b1df83b.pth',
86-
test_input_size=(3, 288, 288), crop_pct=0.95, pool_size=(9, 9)),
86+
input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95),
8787

8888
"lcnet_035": _cfg(),
8989
"lcnet_050": _cfg(),

0 commit comments

Comments
 (0)