使用 NVIDIA TAO 工具套件和預(yù)訓(xùn)練模型加快 AI 開發(fā)

NVIDIA 發(fā)布了 TAO 工具套件 4.0。該工具套件通過全新的 AutoML 功能、與第三方 MLOPs 服務(wù)的集成以及新的預(yù)訓(xùn)練視覺 AI 模型提高開發(fā)者的生產(chǎn)力。該工具套件的企業(yè)版現(xiàn)在可以訪問預(yù)訓(xùn)練模型的完整源代碼和模型權(quán)重。

該工具套件能夠高效訓(xùn)練視覺和對話式 AI 模型。由于簡化了復(fù)雜的 AI 模型和深度學(xué)習(xí)框架，即便是不具備 AI 專業(yè)知識的開發(fā)者也可以使用該工具套件來構(gòu)建 AI 模型。通過遷移學(xué)習(xí)，開發(fā)者可以使用自己的數(shù)據(jù)對 NVIDIA 預(yù)訓(xùn)練模型進(jìn)行微調(diào)，無需 AI 方面的專業(yè)知識或大型訓(xùn)練數(shù)據(jù)集就可以優(yōu)化模型推理的吞吐量。

開發(fā)者可以使用 TAO 創(chuàng)建針對特定環(huán)境和場景進(jìn)行優(yōu)化的自定義生產(chǎn)級模型。新增加的一項(xiàng)重要功能可以幫助開發(fā)者在沒有大量數(shù)據(jù)的情況下構(gòu)建物體檢測模型。用例包括檢測裝配線缺陷、將特定短語翻譯成多種語言、管理城市交通等。

開發(fā)者可以在這里下載 TAO 工具套件 4.0：

https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/resources/tao-getting-started

版本亮點(diǎn)

• 訪問預(yù)訓(xùn)練模型的 TAO 源代碼和模型權(quán)重。

• 使用 AutoML 訓(xùn)練高質(zhì)量的模型，避免了手動微調(diào)數(shù)百個參數(shù)的麻煩。

• 在頂尖云提供商的虛擬機(jī)以及 Amazon EKS 或 Azure AKS 等 Kubernetes 服務(wù)上進(jìn)行部署。

• 簡化基礎(chǔ)設(shè)施的管理以及 Google Colab、Google Vertex AI、Microsoft Azure 機(jī)器學(xué)習(xí)等云端機(jī)器學(xué)習(xí)服務(wù)的擴(kuò)展。

• 新的云集成功能以及 W&B、ClearML 等第三方 MLOps 服務(wù)為開發(fā)者和企業(yè)提供經(jīng)過優(yōu)化的 AI 工作流程。

• 與 REST API 集成?？煽焖贅?gòu)建一項(xiàng)新的 AI 服務(wù)或通過 REST API 集成到現(xiàn)有的服務(wù)中。

• 使用基于轉(zhuǎn)換器的新預(yù)訓(xùn)練模型（CitySemSegformer、Peoplenet Transformer）和專門用于零售業(yè)的預(yù)訓(xùn)練模型（RetailObjectDetection、RetailObjectRecognition 和 ReIdentificationNet）。

入門資源

• 訪問 TAO 工具套件入門頁面，查看教學(xué)視頻和快速入門指南：https://developer.nvidia.com/tao-toolkit-get-started

• 在 NGC 下載 TAO 工具套件和預(yù)訓(xùn)練模型：https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/resources/tao-getting-started

新教程

• 閱讀使用 NVIDIA TAO 的全新 AutoML 功能，像 AI 專業(yè)人士一樣訓(xùn)練：https://developer.nvidia.com/blog/training-like-an-ai-pro-using-tao-automl/

• 閱讀使用 NVIDIA TAO 工具套件和 Azure 機(jī)器學(xué)習(xí)創(chuàng)建自定義 AI 模型，了解如何使用 NVIDIA TAO 工具套件和 Azure 機(jī)器學(xué)習(xí)訓(xùn)練和部署自定義 AI 模型：https://developer.nvidia.com/blog/creating-custom-ai-models-using-nvidia-tao-toolkit-with-azure-machine-learning/

• 觀看視頻，在 Google Colab 上嘗試 NVIDIA TAO 工具套件和預(yù)訓(xùn)練模型：https://www.nvidia.cn/on-demand/session/tao2022-toolkit/

使用 TAO 工具套件的解決方案

• 新一代 IT 和商業(yè)服務(wù)商 Trifork 使用 TAO 工具套件加速開發(fā)機(jī)場 AI 行李追蹤解決方案。

• Fingermark 是一家開發(fā)數(shù)據(jù)驅(qū)動型決策解決方案的公司。該公司使用 TAO 工具套件開發(fā)提高工作人員安全性的工業(yè)視覺 AI 解決方案。