Training Toolbox for PyTorch*

Training Toolbox for PyTorch* provides a convenient environment to train Deep Learning models and convert them using OpenVINO™ Toolkit for optimized inference.

Prerequisites

• Ubuntu* 16.04 / 18.04
• Python* 3.6+
• libturbojpeg
• (Optional) OpenVINO™ to export trained models

Models

After installation, you are ready to train your own models, evaluate and use them for prediction.

• Action Recognition
• Eye State Classification
• Face Antispoofing
• Text recognition
• Human Pose Estimation
• Machine Translation
• Object Re-Identification
  • Face Recognition
  • Person Re-Identification
  • Vehicle Re-Identification
• Segmentation of Thoracic Organs
• Super Resolution
• Question Answering