NetsPresso Training Studio

NetsPresso Training Studio is an open-source project that provides various model architectures, training, evaluation, and inference for AI models compatible with the NetsPresso workflow. Using Training Studio, you can create models that support all of NetsPresso’s capabilities, including initial training, compression, conversion to other intermediate representations, and benchmarking on actual devices in the cloud.

Key Features

Training Studio focuses on model compression and device deployment, ensuring that the models adhere to the following criteria:

• Compatible with torch.fx conversion.
• Compressible using the pruning methods provided by NetsPresso.
• Easily deployable across a variety of edge devices.
• Model Training can be used through the PyNetsPresso API. PyNetsPresso Github-Trainer

Model Structure

To offer a wide range of models that meet these requirements in various forms, Training Studio defines models using four key components: full, backbone, neck, and head. This modular approach allows users to configure backbones, necks, and heads as needed. For models that do not easily fit into these three segments, we provide them as full models.

You can find more information on the official NetsPresso Trainer page below.

NetsPresso Trainer Docs