From b3042081b4ecd7d5c6b006c855cecf9475bce17b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 3 Feb 2023 10:58:02 -0800 Subject: [PATCH] Add laion -> in1k fine-tuned base and large_mlp weights for convnext --- timm/models/convnext.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index da951462..5754e120 100644 --- a/timm/models/convnext.py +++ b/timm/models/convnext.py @@ -723,13 +723,19 @@ default_cfgs = generate_default_cfgs({ # CLIP weights, fine-tuned on in1k or in12k + in1k 'convnext_base.clip_laion2b_augreg_ft_in1k': _cfg( - # hf_hub_id='timm/', + hf_hub_id='timm/', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0), 'convnext_base.clip_laiona_augreg_ft_in1k_384': _cfg( - # hf_hub_id='timm/', + 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), + 'convnext_large_mlp.clip_laion2b_augreg_ft_in1k': _cfg( + hf_hub_id='timm/', + mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, + input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0 + ), + # CLIP based weights, original image tower weights and fine-tunes 'convnext_base.clip_laion2b': _cfg(