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

Other Useful Business Software
$300 in Free Credit Towards Top Cloud Services Icon
$300 in Free Credit Towards Top Cloud Services

Build VMs, containers, AI, databases, storage—all in one place.

Start your project in minutes. After credits run out, 20+ products include free monthly usage. Only pay when you're ready to scale.
Get Started
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