Skip to content

Commit e7bd97b

Browse files
committed
Add missing pretrained_cfg for old untrained model
1 parent 17af914 commit e7bd97b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

timm/models/vision_transformer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1737,7 +1737,9 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
17371737
'vit_medium_patch16_gap_384.sw_in12k_ft_in1k': _cfg(
17381738
hf_hub_id='timm/',
17391739
input_size=(3, 384, 384), crop_pct=0.95, crop_mode='squash'),
1740-
'vit_base_patch16_gap_224': _cfg(),
1740+
'vit_betwixt_patch16_gap_256.untrained': _cfg(
1741+
input_size=(3, 256, 256), crop_pct=0.95),
1742+
'vit_base_patch16_gap_224.untrained': _cfg(),
17411743

17421744
# CLIP pretrained image tower and related fine-tuned weights
17431745
'vit_base_patch32_clip_224.laion2b_ft_in12k_in1k': _cfg(

0 commit comments

Comments
 (0)