From 2bd531e03318f6fde7a939ea68b3866daece5272 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 31 Dec 2024 11:00:44 -0800 Subject: [PATCH] Add 384x384 in12k pretrain and finetune for convnext_nano --- timm/models/convnext.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index 25a767ca..47f2bf87 100644 --- a/timm/models/convnext.py +++ b/timm/models/convnext.py @@ -661,6 +661,9 @@ default_cfgs = generate_default_cfgs({ url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/convnext_tiny_hnf_a2h-ab7e9df2.pth', hf_hub_id='timm/', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0), + 'convnext_nano.r384_in12k_ft_in1k': _cfg( + hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0), 'convnext_tiny.in12k_ft_in1k_384': _cfg( hf_hub_id='timm/', @@ -672,6 +675,12 @@ default_cfgs = generate_default_cfgs({ 'convnext_nano.in12k': _cfg( hf_hub_id='timm/', crop_pct=0.95, num_classes=11821), + 'convnext_nano.r384_in12k': _cfg( + hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, num_classes=11821), + 'convnext_nano.r384_ad_in12k': _cfg( + hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, num_classes=11821), 'convnext_tiny.in12k': _cfg( hf_hub_id='timm/', crop_pct=0.95, num_classes=11821),