OpenVINO™ Training Extensions provide a convenient environment to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference. When ote_cli is installed in the virtual environment, you can use the ote command line interface to perform various actions for templates related to the chosen task type, such as running, training, evaluating, exporting, etc. ote train trains a model (a particular model template) on a dataset and saves results in two files. ote optimize optimizes a pre-trained model using NNCF or POT depending on the model format. NNCF optimization used for trained snapshots in a framework-specific format. POT optimization used for models exported in the OpenVINO IR format.

Features

  • Requires Ubuntu 18.04 / 20.04
  • Supports Python 3.8+
  • Requires CUDA Toolkit 11.1 - for training on GPU
  • The project files can be found in OpenVINO™ Training Extensions
  • Deep Learning Deployment Toolkit is licensed under Apache License Version 2.0
  • Training, export, and evaluation scripts for TensorFlow- and most PyTorch-based models

Project Samples

Project Activity

See All Activity >

License

Apache License V2.0

Follow OpenVINO Training Extensions

OpenVINO Training Extensions Web Site

You Might Also Like
MongoDB Atlas runs apps anywhere Icon
MongoDB Atlas runs apps anywhere

Deploy in 115+ regions with the modern database for every enterprise.

MongoDB Atlas gives you the freedom to build and run modern applications anywhere—across AWS, Azure, and Google Cloud. With global availability in over 115 regions, Atlas lets you deploy close to your users, meet compliance needs, and scale with confidence across any geography.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of OpenVINO Training Extensions!

Additional Project Details

Programming Language

Python

Related Categories

Python Machine Learning Software, Python Deep Learning Frameworks, Python LLM Inference Tool

Registered

2022-08-23