Bring Your Own Model (BYOM)

In TAO 5.0.0, BYOM with TF1 (Classification and UNet) has been deprecated because the source code of TAO Toolkit is now fully open-sourced. To use BYOM with TF1, you will need to continue using TAO 4.0.

Classification TF2 still supports BYOM with the same workflow as TAO 4.0. If you wish to bring your own model weights in TAO 5.0.0, you can directly modify the source code to load the weights.

• BYOM Converter
  
  • Installing the TAO BYOM Converter for TF1 Classification and UNet
  • Installing the TAO BYOM Converter for TF2 Classification
  • Preparing the ONNX Model
  • Running the TAO BYOM Converter
  • Examples of Converting Open-Source Models through TAO BYOM
  • Supported ONNX nodes in TAO BYOM
• BYOM Image Classification