Function nextvit_base

dmidas/backbones/next_vit.py:442–444

(pretrained=False, pretrained_cfg=None, **kwargs)

@register_model
def nextvit_base(pretrained=False, pretrained_cfg=None, **kwargs):
    model = NextViT(stem_chs=[64, 32, 64], depths=[3, 4, 20, 3], path_dropout=0.2, **kwargs)
    return model


@register_model