In this video I will show you how to use the Nvidia TAO toolkit
NVIDIA TAO is a low-code AI toolkit built on TensorFlow and PyTorch, which simplifies and accelerates the model training process by abstracting away the complexity of AI models and the deep learning framework. With TAO, users can select one of 100+ pre-trained vision AI models from NGC and fine-tune and customize on their own dataset without writing a single line of code. The output of TAO is a trained model in ONNX format that can be deployed on any platform that supports ONNX.
Code:
https://github.com/olonok69/LLM_Notebooks/blob/main/ml_Solutions/nvidia/TAO/TAO_multitask_classification.ipynb
Download
0 formats
No download links available.
NVIDIA AI: Nvidia TAO Toolkit Train, Adapt and Optimize #datascience #machinelearning | NatokHD