mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

RepViT weights on HF hub

Browse Source
This commit is contained in:
Ross Wightman 2023-08-25 10:39:02 -07:00
parent 56c285445c
commit 5d599a6a10
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
timm/models/repvit.py
View File

@ -368,13 +368,16 @@ def _cfg(url='', **kwargs):
default_cfgs = generate_default_cfgs( default_cfgs = generate_default_cfgs(
{ {
'repvit_m1.dist_in1k': _cfg( 'repvit_m1.dist_in1k': _cfg(
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_distill_300_timm.pth' hf_hub_id='timm/',
# url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_distill_300_timm.pth'
), ),
'repvit_m2.dist_in1k': _cfg( 'repvit_m2.dist_in1k': _cfg(
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_distill_300_timm.pth' hf_hub_id='timm/',
# url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_distill_300_timm.pth'
), ),
'repvit_m3.dist_in1k': _cfg( 'repvit_m3.dist_in1k': _cfg(
url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m3_distill_300_timm.pth' hf_hub_id='timm/',
# url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m3_distill_300_timm.pth'
), ),
} }
) )