Documentation
Model creation and training
Supported frameworks

Supported frameworks

This section provides an overview of supported model-to-hardware configurations for different machine learning frameworks. You can switch between Classification and Detection models and check compatibility with popular frameworks such as:

  • TensorFlow 2.5.1 Keras,
  • Pytorch 1.8.1,
  • Pytorch 2.2.2,
  • Darknet,
  • TensorFlow 2.11,
  • TensorFlow 2.18 KerasCV.

Each pretrained model can be run on different hardware. The table indicates which combinations are supported. A green supported label indicates that the configuration is supported, while a red unsupported label indicates that it is not.