Training Toolbox for PyTorch* provides a convenient environment to train Deep Learning models and convert them using OpenVINO™ Toolkit for optimized inference.
- Ubuntu* 16.04 / 18.04
- Python* 3.6+
- libturbojpeg
- (Optional) OpenVINO™ to export trained models
After installation, you are ready to train your own models, evaluate and use them for prediction.