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

    - #paths : [2, 3, 3, 3]
    - #layers : [1, 3, 6, 3]
    - #channels : [64, 128, 216, 288]
    - MLP_ratio : 4
    Number of params : 22892400
    FLOPs : 4799650824
    Activations : 30601880
    """

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

- #paths : [2, 3, 3, 3]
- #layers : [1, 3, 6, 3]
- #channels : [64, 128, 216, 288]
- MLP_ratio : 4
Number of params : 22892400
FLOPs : 4799650824
Activations : 30601880
towhee/models/mpvit/mpvit.py