From 9c11dfd9cbca49bca92d207619f3f31abb62d71f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 7 Jan 2022 14:09:50 -0800 Subject: [PATCH] Fix fbnetv3 pretrained cfg changes --- timm/models/mobilenetv3.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 61a7e374..4b9bcc09 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -77,13 +77,13 @@ default_cfgs = { 'fbnetv3_b': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_b_224-ead5d2a1.pth', - test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)), + test_input_size=(3, 256, 256), crop_pct=0.95), 'fbnetv3_d': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_d_224-c98bce42.pth', - test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)), + test_input_size=(3, 256, 256), crop_pct=0.95), 'fbnetv3_g': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_g_240-0b1df83b.pth', - test_input_size=(3, 288, 288), crop_pct=0.95, pool_size=(9, 9)), + input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95), "lcnet_035": _cfg(), "lcnet_050": _cfg(),