diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index f2cd8f34..8e7ca905 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -2352,8 +2352,8 @@ default_cfgs = generate_default_cfgs({ input_size=(3, 256, 256), pool_size=(8, 8), ), - 'test_byobnet.untrained': _cfgr( - # hf_hub_id='timm/', + 'test_byobnet.r160_in1k': _cfgr( + hf_hub_id='timm/', first_conv='stem.conv', input_size=(3, 160, 160), crop_pct=0.875, pool_size=(5, 5), ), diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 289c83f3..56bd3707 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1756,8 +1756,8 @@ default_cfgs = generate_default_cfgs({ #hf_hub_id='timm/', input_size=(3, 224, 224), crop_pct=0.9), - "test_efficientnet.untrained": _cfg( - # hf_hub_id='timm/' + "test_efficientnet.r160_in1k": _cfg( + hf_hub_id='timm/', input_size=(3, 160, 160), pool_size=(5, 5)), }) diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 08c04d54..4728614a 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -1990,7 +1990,8 @@ default_cfgs = { 'vit_so150m_patch16_reg4_map_256.untrained': _cfg( 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), }