Add 384x384 convnext_large_mlp laion2b fine-tune on in1k

2 years ago · 0d33127df2
parent 88a5b8491d
commit 0d33127df2
1 changed files with 5 additions and 0 deletions
--- a/timm/models/convnext.py
+++ b/timm/models/convnext.py
@ -735,6 +735,11 @@ default_cfgs = generate_default_cfgs({
        mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
        input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0
    ),
+    'convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384': _cfg(
+        hf_hub_id='timm/',
+        mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
+        input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'
+    ),


    # CLIP based weights, original image tower weights and fine-tunes