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

• OTE SDK
• OTE CLI
• OTE Algorithms

Quick start guide

In order to get started with OpenVINO™ Training Extensions see the quick-start guide.

GitHub Repository

The project files can be found in OpenVINO™ Training Extensions. Previously developed models can be found on the misc branch.

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.

Contributing

Please read the Contribution guide before starting work on a pull request.

Known limitations

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.