From 884a0f1a125733c98787a2f7696687b13aeff2a4 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 10 Nov 2022 00:06:57 -0800 Subject: [PATCH] Add openai b32 ft --- timm/models/vision_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 52a7feeb..d4d091da 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -855,7 +855,7 @@ default_cfgs = generate_defaults({ mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0, num_classes=768), 'vit_base_patch32_clip_224.openai_ft_in1k': _cfg( - #hf_hub_id='timm/vit_base_patch32_clip_224.openai_ft_in1k', + hf_hub_id='timm/vit_base_patch32_clip_224.openai_ft_in1k', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD), 'vit_base_patch16_clip_224.openai_ft_in1k': _cfg( hf_hub_id='timm/vit_base_patch16_clip_224.openai_ft_in1k',