[NVIDIA TAO Toolkit] TAO(Train, Adapt, and Optimize) Toolkit
NVIDIA TAO Toolkit User Guide : https://docs.nvidia.com/tao/tao-toolkit/index.html
TAO Toolkit — TAO Toolkit 3.21.11 documentation
© Copyright 2021, NVIDIA. Last updated on Nov 29, 2021.
docs.nvidia.com
NVIDIA TAO Toolkit Release Notes : https://docs.nvidia.com/tao/tao-toolkit/text/release_notes.html
Release Notes — TAO Toolkit 3.21.11 documentation
Release Contents Components included in this release: TAO Launcher pip package TAO - TF docker TAO - Pytorch Docker TAO - Language Model Docker Jupyter notebook with sample workflows
docs.nvidia.com
NVIDIA에서 제공하는 TAO(Train, Adapt, and Optimize) Toolkit 은 자신의 데이터로 pretrained model을 fine-turning하고 edge device를 통해 TensorRT 기반 inference를 위해 export 할 수 있는 Python package이다. DeepStream과 같이 사용하여 모델을 학습하고, 배포하기까지 손쉽게 할 수 있다. Object Detection 부터 시작해서 여러 딥러닝 모델들을 제공하고 있기 때문에 매우 유용할 것 같다.
참고로 TAO 라는 Toolkit이 새로 나온줄 알았는데 기존에 알고 있었던 TLT(Transfer Learning Toolkit)였다. 😊
NVIDIA Transfer Learning Toolkit has been renamed to TAO Toolkit.
GUI 버전도 early access 할 수 있게 제공하고 있으며, 참고로 조직의 이메일 주소를 사용해야 access 할 수 있다.
https://developer.nvidia.com/tao-early-access
NVIDIA TAO (GUI VERSION) - EARLY ACCESS
NVIDIA TAO (GUI VERSION) - EARLY ACCESS NVIDIA TAO is an AI-model-adaptation framework that simplifies and accelerates the creation of enterprise AI applications and services. This is an early access program for the GUI-based version of TAO available to a
developer.nvidia.com
TAO는 학습이 가능하고 즉시 배포 가능한 모델들을 제공하고 있다.
이 포스팅을 시작으로 TAO Toolkit을 사용하는 방법을 차근차근 다루어보려고 한다.
'AI Development > TensorRT' 카테고리의 다른 글
| [NVIDIA TAO Toolkit] TAO Toolkit 개요 (1) | 2022.05.03 |
|---|---|
| [TensorRT] trtexec dumpProfile (0) | 2022.03.22 |
| [TensorRT] Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32. (0) | 2022.02.23 |
| [TensorRT] ONNX 및 TRT에서 Group Normalization 사용하기 (+ Instance Normalization 이슈) (0) | 2022.02.23 |
| [TensorRT] QuickStartGuide (6) | 2021.08.31 |