Step 6: Retrain model

Currently, the scope of retraining specified in NetsPresso aims at restoring the performance of the model after compression.

The results after retraining are saved in the Trained Model directory. As mentioned in the Model Saved Directory above, a directory with the "Re" prefix will be created under the target model, and the results will be stored in that directory.

Retraining Configuration

• Batch Size
• Epochs
• Optimizer
• Learning Rate (Optional)
• GPU (Optional)