OpenVINO™ Training Extensions
OpenVINO™ Training Extensions provide a convenient environment to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference.
Prerequisites
- Ubuntu 18.04 / 20.04
- Python 3.8+
- CUDA Toolkit 11.1 - for training on GPU
Repository components
Quick start guide
In order to get started with OpenVINO™ Training Extensions click here.
License
Deep Learning Deployment Toolkit is licensed under Apache License Version 2.0. By contributing to the project, you agree to the license and copyright terms therein and release your contribution under these terms.
Misc
Models that were previously developed can be found on the misc branch.
Contributing
Please read the Contributing guide before starting work on a pull request.
Known limitations
Currently, training, exporting, evaluation scripts for TensorFlow*-based models and the most of PyTorch*-based models from the misc branch are exploratory and are not validated.
* Other names and brands may be claimed as the property of others.