Home

NetsPresso Trainer is a PyTorch training repository specialized for model training with torch.fx graphmodule conversion and automatic compression.

By using NetsPresso Trainer and NetsPresso together, you can take advantage of features such as model compression without significant performance drop, ONNX export, and NVIDIA TensorRT export.

Enjoy training and deploying your own edge AI models with NetsPresso Trainer!