Skip to content

Commit 462fb3e

Browse files
committed
Push new repvit weights to hub, tweak tag names
1 parent 5309424 commit 462fb3e

File tree

1 file changed

+20
-20
lines changed

1 file changed

+20
-20
lines changed

timm/models/repvit.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -394,35 +394,35 @@ def _cfg(url='', **kwargs):
394394
'repvit_m3.dist_in1k': _cfg(
395395
hf_hub_id='timm/',
396396
),
397-
'repvit_m0_9.dist_in1k_300e': _cfg(
398-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m0_9_distill_300e_timm.pth'
397+
'repvit_m0_9.dist_300e_in1k': _cfg(
398+
hf_hub_id='timm/',
399399
),
400-
'repvit_m0_9.dist_in1k_450e': _cfg(
401-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m0_9_distill_450e_timm.pth'
400+
'repvit_m0_9.dist_450e_in1k': _cfg(
401+
hf_hub_id='timm/',
402402
),
403-
'repvit_m1_0.dist_in1k_300e': _cfg(
404-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_0_distill_300e_timm.pth'
403+
'repvit_m1_0.dist_300e_in1k': _cfg(
404+
hf_hub_id='timm/',
405405
),
406-
'repvit_m1_0.dist_in1k_450e': _cfg(
407-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_0_distill_450e_timm.pth'
406+
'repvit_m1_0.dist_450e_in1k': _cfg(
407+
hf_hub_id='timm/',
408408
),
409-
'repvit_m1_1.dist_in1k_300e': _cfg(
410-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_1_distill_300e_timm.pth'
409+
'repvit_m1_1.dist_300e_in1k': _cfg(
410+
hf_hub_id='timm/',
411411
),
412-
'repvit_m1_1.dist_in1k_450e': _cfg(
413-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_1_distill_450e_timm.pth'
412+
'repvit_m1_1.dist_450e_in1k': _cfg(
413+
hf_hub_id='timm/',
414414
),
415-
'repvit_m1_5.dist_in1k_300e': _cfg(
416-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_5_distill_300e_timm.pth'
415+
'repvit_m1_5.dist_300e_in1k': _cfg(
416+
hf_hub_id='timm/',
417417
),
418-
'repvit_m1_5.dist_in1k_450e': _cfg(
419-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_5_distill_450e_timm.pth'
418+
'repvit_m1_5.dist_450e_in1k': _cfg(
419+
hf_hub_id='timm/',
420420
),
421-
'repvit_m2_3.dist_in1k_300e': _cfg(
422-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_3_distill_300e_timm.pth'
421+
'repvit_m2_3.dist_300e_in1k': _cfg(
422+
hf_hub_id='timm/',
423423
),
424-
'repvit_m2_3.dist_in1k_450e': _cfg(
425-
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_3_distill_450e_timm.pth'
424+
'repvit_m2_3.dist_450e_in1k': _cfg(
425+
hf_hub_id='timm/',
426426
),
427427
}
428428
)

0 commit comments

Comments
 (0)