From 0d33127df2e7c0e4476662eaa719799b9f75f4c5 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 6 Feb 2023 22:01:04 -0800 Subject: [PATCH] Add 384x384 convnext_large_mlp laion2b fine-tune on in1k --- timm/models/convnext.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index 5754e120..ce7b5a27 100644 --- 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