Skip to content

Commit 84cb225

Browse files
committed
Add in12k + 12k_ft_in1k vit_medium weights
1 parent 4634c3e commit 84cb225

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

timm/models/vision_transformer.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1878,9 +1878,16 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
18781878
'vit_medium_patch16_reg1_gap_256.sbb_in1k': _cfg(
18791879
hf_hub_id='timm/',
18801880
input_size=(3, 256, 256), crop_pct=0.95),
1881+
'vit_medium_patch16_reg4_gap_256.sbb_in12k_ft_in1k': _cfg(
1882+
hf_hub_id='timm/',
1883+
input_size=(3, 256, 256), crop_pct=0.95),
18811884
'vit_medium_patch16_reg4_gap_256.sbb_in1k': _cfg(
18821885
hf_hub_id='timm/',
18831886
input_size=(3, 256, 256), crop_pct=0.95),
1887+
'vit_medium_patch16_reg4_gap_256.sbb_in12k': _cfg(
1888+
hf_hub_id='timm/',
1889+
num_classes=11821,
1890+
input_size=(3, 256, 256), crop_pct=0.95),
18841891
'vit_mediumd_patch16_reg4_gap_256.sbb_in12k_ft_in1k': _cfg(
18851892
hf_hub_id='timm/',
18861893
input_size=(3, 256, 256), crop_pct=0.95),

0 commit comments

Comments
 (0)