mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Comment out time local files while testing new vit weights

pull/2169/head
Ross Wightman 2024-05-03 20:26:56 -07:00
parent 3e7ab12af9
commit f8979d4f50
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
timm/models/eva.py
View File

@ -953,24 +953,24 @@ default_cfgs = generate_default_cfgs({
'vit_medium_patch16_rope_reg1_gap_256.in1k': _cfg(
#hf_hub_id='timm/',
file='vit_medium_gap1_rope-in1k-20230920-5.pth',
#file='vit_medium_gap1_rope-in1k-20230920-5.pth',
input_size=(3, 256, 256), crop_pct=0.95,
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
),
'vit_mediumd_patch16_rope_reg1_gap_256.in1k': _cfg(
#hf_hub_id='timm/',
file='vit_mediumd_gap1_rope-in1k-20230926-5.pth',
#file='vit_mediumd_gap1_rope-in1k-20230926-5.pth',
input_size=(3, 256, 256), crop_pct=0.95,
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
),
'vit_betwixt_patch16_rope_reg4_gap_256.in1k': _cfg(
#hf_hub_id='timm/',
file='vit_betwixt_gap4_rope-in1k-20231005-5.pth',
#file='vit_betwixt_gap4_rope-in1k-20231005-5.pth',
input_size=(3, 256, 256), crop_pct=0.95,
),
'vit_base_patch16_rope_reg1_gap_256.in1k': _cfg(
# hf_hub_id='timm/',
file='vit_base_gap1_rope-in1k-20230930-5.pth',
#hf_hub_id='timm/',
#file='vit_base_gap1_rope-in1k-20230930-5.pth',
input_size=(3, 256, 256), crop_pct=0.95,
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
),

12
timm/models/vision_transformer.py
View File

@ -1791,22 +1791,22 @@ default_cfgs = {
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=512),
'vit_wee_patch16_reg1_gap_256': _cfg(
file='',
#file='',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_little_patch16_reg4_gap_256': _cfg(
file='',
#file='',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_medium_patch16_reg1_gap_256': _cfg(
file='vit_medium_gap1-in1k-20231118-8.pth',
#file='vit_medium_gap1-in1k-20231118-8.pth',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_medium_patch16_reg4_gap_256': _cfg(
file='vit_medium_gap4-in1k-20231115-8.pth',
#file='vit_medium_gap4-in1k-20231115-8.pth',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_betwixt_patch16_reg1_gap_256': _cfg(
file='vit_betwixt_gap1-in1k-20231121-8.pth',
#file='vit_betwixt_gap1-in1k-20231121-8.pth',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_betwixt_patch16_reg4_gap_256': _cfg(
file='vit_betwixt_gap4-in1k-20231106-8.pth',
#file='vit_betwixt_gap4-in1k-20231106-8.pth',
input_size=(3, 256, 256), crop_pct=0.95),
'vit_base_patch16_reg4_gap_256': _cfg(
input_size=(3, 256, 256)),