Skip to content

Overview

NetsPresso Trainer and NetsPresso service provide a convenient experience in training, compressing, retraining, and deploying models for user devices, seamlessly. In that process, NetsPresso Trainer manages both training and retraining phases, ensuring the models are fully compatible with NetsPresso.

NetsPresso Trainer categorizes essential parameters for training into six configuration modules. Each module is responsible for the following aspects:

  • Data: Defines the structure of the user-customized or Hugging Face datasets for interpretation by NetsPresso Trainer.
  • Augmentation: Defines the data augmentation recipe.
  • Model: Defines the model architecture, loss modules, and pretrained weights.
  • Training: Defines necessary elements like optimizer, epochs, and batch size for training.
  • Logging: Defines output formats of training results.
  • Environment: Defines the training environment, including GPU usage and dataloader multi-processing.

This component section describes in detail the six configuration modules which are necessary to use NetsPresso Trainer. You can see yaml configuration examples in our public repository.

Advantage of NetsPresso Trainer

Use SOTA models fully compatible with NetsPresso

NetsPresso Trainer provides reimplemented SOTA models that ensure compatibility with NetsPresso. This allows users to avoid expending resources on changing model formats for model compression and device deployment. Therefore, the users can easily utilize SOTA models to their applications.

Easily trainable with yaml configuration

NetsPresso Trainer encapsulates all the necessary values within configurations for model training. This enables extensive optimization attempts with mere modifications of these configuration files. Also, this enhances usability by using same configuration format for retraining compressed models.