From a163411a67d389fb7dd34df08c29a19a91046e05 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 22 Nov 2024 09:25:55 -0800 Subject: [PATCH 1/2] CS3-DarkNet Small (Focus) w/ RA4 recipe. Fix #2122 --- timm/models/cspnet.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/timm/models/cspnet.py b/timm/models/cspnet.py index a736882190..81d11a0654 100644 --- a/timm/models/cspnet.py +++ b/timm/models/cspnet.py @@ -963,7 +963,10 @@ def _cfg(url='', **kwargs): url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_x_c2ns-4e4490aa.pth', interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0), - 'cs3darknet_focus_s.untrained': _cfg(interpolation='bicubic'), + 'cs3darknet_focus_s.ra4_e3600_r256_in1k': _cfg( + hf_hub_id='timm/', + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), + interpolation='bicubic', test_input_size=(3, 320, 320), test_crop_pct=1.0), 'cs3darknet_focus_m.c2ns_in1k': _cfg( hf_hub_id='timm/', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_focus_m_c2ns-e23bed41.pth', From 54bdab7411d2c5a3d6ee743cb9e20ea10ab42c4c Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 22 Nov 2024 15:24:53 -0800 Subject: [PATCH 2/2] add mnv4 conv_medium in12k -> in1k ft --- timm/models/mobilenetv3.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 16d5dc7016..c5e81391f2 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1032,6 +1032,10 @@ def _cfg(url: str = '', **kwargs): hf_hub_id='timm/', crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'), + 'mobilenetv4_conv_medium.e250_r384_in12k_ft_in1k': _cfg( + hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), + crop_pct=0.95, interpolation='bicubic'), 'mobilenetv4_conv_medium.e180_r384_in12k': _cfg( hf_hub_id='timm/', num_classes=11821,