使用NVIDIA TAO工具套件和预训练模型加快AI开发

NVIDIA发布了TAO工具套件4.0。该工具套件通过全新的AutoML功能、与第三方MLOPs服务的集成以及新的预训练视觉AI模型提高开发者的生产力。该工具套件的企业版现在可以访问预训练模型的完整源代码和模型权重。

该工具套件能够高效训练视觉和对话式AI模型。由于简化了复杂的AI模型和深度学习框架，即便是不具备AI专业知识的开发者也可以使用该工具套件来构建AI模型。通过迁移学习，开发者可以使用自己的数据对NVIDIA预训练模型进行微调，无需AI方面的专业知识或大型训练数据集就可以优化模型推理的吞吐量。

开发者可以使用TAO创建针对特定环境和场景进行优化的自定义生产级模型。新增加的一项重要功能可以帮助开发者在没有大量数据的情况下构建物体检测模型。用例包括检测装配线缺陷、将特定短语翻译成多种语言、管理城市交通等。

开发者可以在这里下载TAO工具套件4.0。

版本亮点

• 访问预训练模型的TAO源代码和模型权重。
• 使用AutoML训练高质量的模型，避免了手动微调数百个参数的麻烦。
• 在顶尖云提供商的虚拟机以及Amazon EKS或 Azure AKS等Kubernetes服务上进行部署。
• 简化基础设施的管理以及Google Colab、Google Vertex AI、Microsoft Azure机器学习等云端机器学习服务的扩展。
• 新的云集成功能以及W&B、ClearML等第三方MLOps服务为开发者和企业提供经过优化的AI工作流程。
• 与REST API集成。可快速构建一项新的AI服务或通过REST API集成到现有的服务中。
• 使用基于转换器的新预训练模型（CitySemSegformer、Peoplenet Transformer）和专门用于零售业的预训练模型（RetailObjectDetection、RetailObjectRecognition和ReIdentificationNet）。

入门资源

• 访问TAO 工具套件入门页面，查看教学视频和快速入门指南。
• 在NGC下载TAO工具套件和预训练模型。

新教程

• 阅读使用NVIDIA TAO的全新AutoML功能，像AI专业人士一样训练。
• 阅读使用NVIDIA TAO工具套件和Azure机器学习创建自定义AI模型，了解如何使用NVIDIA TAO工具套件和Azure机器学习训练和部署自定义AI模型。
• 观看视频，在Google Colab上尝试NVIDIA TAO工具套件和预训练模型。

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

• 新一代IT和商业服务商Trifork使用TAO工具套件加速开发机场AI行李追踪解决方案。
• Fingermark是一家开发数据驱动型决策解决方案的公司。该公司使用TAO工具套件开发提高工作人员安全性的工业视觉AI解决方案。