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

Add weights for test models

Browse Source
This commit is contained in:
Ross Wightman 2024-07-22 10:08:57 -07:00
parent 1a05ed29a1
commit 9b2b8014e8
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/byobnet.py
View File

@ -2352,8 +2352,8 @@ default_cfgs = generate_default_cfgs({
input_size=(3, 256, 256), pool_size=(8, 8), input_size=(3, 256, 256), pool_size=(8, 8),
), ),
'test_byobnet.untrained': _cfgr( 'test_byobnet.r160_in1k': _cfgr(
# hf_hub_id='timm/', hf_hub_id='timm/',
first_conv='stem.conv', first_conv='stem.conv',
input_size=(3, 160, 160), crop_pct=0.875, pool_size=(5, 5), input_size=(3, 160, 160), crop_pct=0.875, pool_size=(5, 5),
), ),

4
timm/models/efficientnet.py
View File

@ -1756,8 +1756,8 @@ default_cfgs = generate_default_cfgs({
#hf_hub_id='timm/', #hf_hub_id='timm/',
input_size=(3, 224, 224), crop_pct=0.9), input_size=(3, 224, 224), crop_pct=0.9),
"test_efficientnet.untrained": _cfg( "test_efficientnet.r160_in1k": _cfg(
# hf_hub_id='timm/' hf_hub_id='timm/',
input_size=(3, 160, 160), pool_size=(5, 5)), input_size=(3, 160, 160), pool_size=(5, 5)),
}) })

3
timm/models/vision_transformer.py
View File

@ -1990,7 +1990,8 @@ default_cfgs = {
'vit_so150m_patch16_reg4_map_256.untrained': _cfg( 'vit_so150m_patch16_reg4_map_256.untrained': _cfg(
input_size=(3, 256, 256)), input_size=(3, 256, 256)),
'test_vit.untrained': _cfg( 'test_vit.r160_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 160, 160), crop_pct=0.875), input_size=(3, 160, 160), crop_pct=0.875),
} }