|
|
@ -236,6 +236,12 @@ def vit_base_r50_s16_384(pretrained=False, **kwargs):
|
|
|
|
return model
|
|
|
|
return model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register_model
|
|
|
|
|
|
|
|
def vit_base_resnet50_384(pretrained=False, **kwargs):
|
|
|
|
|
|
|
|
# DEPRECATED this is forwarding to model def above for backwards compatibility
|
|
|
|
|
|
|
|
return vit_base_r50_s16_384(pretrained=pretrained, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register_model
|
|
|
|
@register_model
|
|
|
|
def vit_large_r50_s32_224(pretrained=False, **kwargs):
|
|
|
|
def vit_large_r50_s32_224(pretrained=False, **kwargs):
|
|
|
|
""" R50+ViT-L/S32 hybrid.
|
|
|
|
""" R50+ViT-L/S32 hybrid.
|
|
|
@ -292,6 +298,12 @@ def vit_base_r50_s16_224_in21k(pretrained=False, **kwargs):
|
|
|
|
return model
|
|
|
|
return model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register_model
|
|
|
|
|
|
|
|
def vit_base_resnet50_224_in21k(pretrained=False, **kwargs):
|
|
|
|
|
|
|
|
# DEPRECATED this is forwarding to model def above for backwards compatibility
|
|
|
|
|
|
|
|
return vit_base_r50_s16_224_in21k(pretrained=pretrained, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register_model
|
|
|
|
@register_model
|
|
|
|
def vit_large_r50_s32_224_in21k(pretrained=False, **kwargs):
|
|
|
|
def vit_large_r50_s32_224_in21k(pretrained=False, **kwargs):
|
|
|
|
""" R50+ViT-L/S32 hybrid. ImageNet-21k.
|
|
|
|
""" R50+ViT-L/S32 hybrid. ImageNet-21k.
|
|
|
|