Available Models

NetsPresso Trainer supports the following model structures.

Full Models

• PIDNet

Backbones

• CSPDarkNet
• EfficientFormer
• MixNet
• MixTransformer
• MobileNetV3
• MobileNetV4
• MobileViT
• ResNet
• ViT

Necks

• FPN
• YOLOPAFPN

Heads

• AllMLPDecoder
• AnchorDecoupledHead
• AnchorFreeDecoupledHead
• FC
• RTMCC

Layers

• Activations
• Normalizations