Add openai b32 ft

2 years ago · 884a0f1a12
parent 2c80da3b9a
commit 884a0f1a12
1 changed files with 1 additions and 1 deletions
--- 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',