Skip to content

Commit 381b279

Browse files
committed
Add hybrid model fwds back
1 parent 26f04a8 commit 381b279

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

tests/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ def test_model_default_cfgs_non_std(model_name, batch_size):
173173
state_dict = model.state_dict()
174174
cfg = model.default_cfg
175175

176-
input_size = _get_input_size(model_name=model_name, target=TARGET_FWD_SIZE)
176+
input_size = _get_input_size(model=model)
177177
if max(input_size) > 320: # FIXME const
178178
pytest.skip("Fixed input size model > limit.")
179179

timm/models/vision_transformer_hybrid.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,12 @@ def vit_base_r50_s16_384(pretrained=False, **kwargs):
236236
return model
237237

238238

239+
@register_model
240+
def vit_base_resnet50_384(pretrained=False, **kwargs):
241+
# DEPRECATED this is forwarding to model def above for backwards compatibility
242+
return vit_base_r50_s16_384(pretrained=pretrained, **kwargs)
243+
244+
239245
@register_model
240246
def vit_large_r50_s32_224(pretrained=False, **kwargs):
241247
""" R50+ViT-L/S32 hybrid.
@@ -292,6 +298,12 @@ def vit_base_r50_s16_224_in21k(pretrained=False, **kwargs):
292298
return model
293299

294300

301+
@register_model
302+
def vit_base_resnet50_224_in21k(pretrained=False, **kwargs):
303+
# DEPRECATED this is forwarding to model def above for backwards compatibility
304+
return vit_base_r50_s16_224_in21k(pretrained=pretrained, **kwargs)
305+
306+
295307
@register_model
296308
def vit_large_r50_s32_224_in21k(pretrained=False, **kwargs):
297309
""" R50+ViT-L/S32 hybrid. ImageNet-21k.

0 commit comments

Comments
 (0)