From 26f04a8e3ef7c581f4766cd34e71d94105c32064 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 19 Jun 2021 16:39:36 -0700 Subject: [PATCH] Fix a weight link --- timm/models/vision_transformer_hybrid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/vision_transformer_hybrid.py b/timm/models/vision_transformer_hybrid.py index a53419a0..30330d39 100644 --- a/timm/models/vision_transformer_hybrid.py +++ b/timm/models/vision_transformer_hybrid.py @@ -69,7 +69,7 @@ default_cfgs = { # hybrid in-21k models (weights from official Google JAX impl where they exist) 'vit_tiny_r_s16_p8_224_in21k': _cfg( - url='https://storage.googleapis.com/vit_models/augreg/Ti_16-i1k-300ep-lr_0.001-aug_medium1-wd_0.1-do_0.0-sd_0.0.npz', + url='https://storage.googleapis.com/vit_models/augreg/R_Ti_16-i21k-300ep-lr_0.001-aug_none-wd_0.03-do_0.0-sd_0.0.npz', num_classes=21843, crop_pct=0.9, first_conv='patch_embed.backbone.conv'), 'vit_small_r26_s32_224_in21k': _cfg( url='https://storage.googleapis.com/vit_models/augreg/R26_S_32-i21k-300ep-lr_0.001-aug_medium2-wd_0.03-do_0.0-sd_0.0.npz',