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

    - #paths : [2, 3, 3, 3]
    - #layers : [1, 3, 8, 3]
    - #channels : [128, 224, 368, 480]
    MLP_ratio : 4
    Number of params: 74845976
    FLOPs : 16445326240
    Activations : 60204392
    """

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

- #paths : [2, 3, 3, 3]
- #layers : [1, 3, 8, 3]
- #channels : [128, 224, 368, 480]
MLP_ratio : 4
Number of params: 74845976
FLOPs : 16445326240
Activations : 60204392
towhee/models/mpvit/mpvit.py