def mpvit_xsmall(**kwargs) -> MPViT:
    """mpvit_xsmall :

    - #paths : [2, 3, 3, 3]
    - #layers : [1, 2, 4, 1]
    - #channels : [64, 128, 192, 256]
    - MLP_ratio : 4
    Number of params : 10573448
    FLOPs : 2971396560
    Activations : 21983464
    """

    model = MPViT(
        img_size=224,
        num_stages=4,
        num_path=[2, 3, 3, 3],
        num_layers=[1, 2, 4, 1],
        embed_dims=[64, 128, 192, 256],
        mlp_ratios=[4, 4, 4, 4],
        num_heads=[8, 8, 8, 8],
        **kwargs,
    )
    model.default_cfg = _cfg_mpvit()
    return model
mpvit_xsmall :

- #paths : [2, 3, 3, 3]
- #layers : [1, 2, 4, 1]
- #channels : [64, 128, 192, 256]
- MLP_ratio : 4
Number of params : 10573448
FLOPs : 2971396560
Activations : 21983464
towhee/models/mpvit/mpvit.py