Skip to content

Commit a2c14c2

Browse files
committed
Add tiny/small in12k pretrained and fine-tuned ConvNeXt models
1 parent 01aea8c commit a2c14c2

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,12 @@ And a big thanks to all GitHub sponsors who helped with some of my costs before
2424
* ❗Updates after Oct 10, 2022 are available in 0.8.x pre-releases (`pip install --pre timm`) or cloning main❗
2525
* Stable releases are 0.6.x and available by normal pip install or clone from [0.6.x](https://github.com/rwightman/pytorch-image-models/tree/0.6.x) branch.
2626

27+
### Jan 11, 2023
28+
* Update ConvNeXt ImageNet-12k pretrain series w/ two new fine-tuned weights (and pre FT `.in12k` tags)
29+
* `convnext_nano.in12k_ft_in1k` - 82.3 @ 224, 82.9 @ 288 (previously released)
30+
* `convnext_tiny.in12k_ft_in1k` - 84.2 @ 224, 84.5 @ 288
31+
* `convnext_small.in12k_ft_in1k` - 85.2 @ 224, 85.3 @ 288
32+
2733
### Jan 6, 2023
2834
* Finally got around to adding `--model-kwargs` and `--opt-kwargs` to scripts to pass through rare args directly to model classes from cmd line
2935
* `train.py /imagenet --model resnet50 --amp --model-kwargs output_stride=16 act_layer=silu`

timm/models/convnext.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -478,10 +478,22 @@ def _cfgv2(url='', **kwargs):
478478
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/convnext_tiny_hnf_a2h-ab7e9df2.pth',
479479
hf_hub_id='timm/',
480480
crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
481+
'convnext_tiny.in12k_ft_in1k': _cfg(
482+
hf_hub_id='timm/',
483+
crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
484+
'convnext_small.in12k_ft_in1k': _cfg(
485+
hf_hub_id='timm/',
486+
crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
481487

482488
'convnext_nano.in12k': _cfg(
483489
hf_hub_id='timm/',
484490
crop_pct=0.95, num_classes=11821),
491+
'convnext_tiny.in12k': _cfg(
492+
hf_hub_id='timm/',
493+
crop_pct=0.95, num_classes=11821),
494+
'convnext_small.in12k': _cfg(
495+
hf_hub_id='timm/',
496+
crop_pct=0.95, num_classes=11821),
485497

486498
'convnext_tiny.fb_in1k': _cfg(
487499
url="https://dl.fbaipublicfiles.com/convnext/convnext_tiny_1k_224_ema.pth",

0 commit comments

Comments
 (0)