OpenVINO™ Training Extensions provide a convenient environment to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference.
- Ubuntu 18.04 / 20.04
- Python 3.8+
- CUDA Toolkit 11.1 - for training on GPU
In order to get started with OpenVINO™ Training Extensions see the quick-start guide.
The project files can be found in OpenVINO™ Training Extensions. Previously developed models can be found on the misc branch.
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.
Please read the Contribution guide before starting work on a pull request.
Training, export, and evaluation scripts for TensorFlow- and most PyTorch-based models from the misc branch are, currently, not production-ready. They serve exploratory purposes and are not validated.
* Other names and brands may be claimed as the property of others.