From 575978ba55cd66a14325641b5d956a8b6083ec2f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 13 Jun 2024 12:57:19 -0700 Subject: [PATCH] Add mnv4_conv_large 384x384 weight location --- timm/models/mobilenetv3.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 5288df6f..196aaceb 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1006,9 +1006,10 @@ default_cfgs = generate_default_cfgs({ 'mobilenetv4_conv_medium.e500_r224_in1k': _cfg( hf_hub_id='timm/', crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'), - 'mobilenetv4_conv_large.r384': _cfg( - # hf_hub_id='timm/', - input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=0.95, interpolation='bicubic'), + 'mobilenetv4_conv_large.e600_r384_in1k': _cfg( + hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), + crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'), 'mobilenetv4_conv_large.e500_r256_in1k': _cfg( hf_hub_id='timm/', input_size=(3, 256, 256), pool_size=(8, 8),