CompressAI Trainer

CompressAI Trainer is a training platform that assists in managing experiments for end-to-end neural network-based compression research.

CompressAI Trainer integrates with:

• CompressAI (library)

• Aim (experiment tracker)

• Catalyst (training engine)

• Hydra (YAML configuration)

These tools give researchers more flexibility, experiment reproducibility, experiment management, and effortless visualizations and metrics.

CompressAI Trainer integrates with the Aim experiment tracker to display live visualizations of RD curves during training.