Skip to content

Commit b304208

Browse files
committed
Add laion -> in1k fine-tuned base and large_mlp weights for convnext
1 parent 316bdf8 commit b304208

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

timm/models/convnext.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -723,13 +723,19 @@ def _cfgv2(url='', **kwargs):
723723

724724
# CLIP weights, fine-tuned on in1k or in12k + in1k
725725
'convnext_base.clip_laion2b_augreg_ft_in1k': _cfg(
726-
# hf_hub_id='timm/',
726+
hf_hub_id='timm/',
727727
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
728728
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0),
729729
'convnext_base.clip_laiona_augreg_ft_in1k_384': _cfg(
730-
# hf_hub_id='timm/',
730+
hf_hub_id='timm/',
731731
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
732732
input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0),
733+
'convnext_large_mlp.clip_laion2b_augreg_ft_in1k': _cfg(
734+
hf_hub_id='timm/',
735+
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
736+
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0
737+
),
738+
733739

734740
# CLIP based weights, original image tower weights and fine-tunes
735741
'convnext_base.clip_laion2b': _cfg(

0 commit comments

Comments
 (0)