使用最新版本的 TAO 工具套件简化 AI 模型开发

利用新的预训练模型以及 ONNX 模型权重导入、REST API 和 TensorBoard 可视化等功能大幅提升生产力和模型训练速度。

TAO 更新.jpg

NVIDIA于6月7日发布了最新版本的 TAO 工具套件。这是 NVIDIA 训练、适应和优化(TAO)框架的低代码版本,它能够简化并加速语音和视觉 AI 应用的 AI 模型创建。

TAO 使开发者能够轻松运用迁移学习创建自定义生产级模型,这些模型专门针对缺陷检测、语言翻译、交通管理等各种行业专属用例进行了优化。使用 TAO 开发模型的用户能够用更少的数据优化模型,进而缩短部署时间。

最新发布的 TAO 工具套件包括全新和更新的视觉及语音预训练模型。ONNX 模型权重导入、REST API 和 TensorBoard 集成等新功能,能够快速追踪模型创建流程,提高开发者的生产力。

版本亮点

使用自己的模型权重:通过导入 ONNX 图像分类和分割模型的预训练权重来加快模型的自定义。您也可以使用您自己的数据自定义和优化模型。

使用 REST API 将 TAO 工具套件作为服务部署:使用 REST API 将 TAO 工具套件作为服务部属到 Kubernetes 上的现代云原生基础设施中,从而构建新的 AI 服务或集成到现有服务中。

使用 TensorBoard 实现可视化:通过训练和验证损失、模型权重、TensorBoard 中的预测图像等可视化标量了解您的模型训练性能。改变超参数,比较实验结果并作出最适合您需求的选择。

预训练模型:预训练模型加快自定义流程,让您能够通过迁移学习以更少的数据进行微调。

最新版本中新增的一些预训练模型可以:

  • 将从激光雷达传感器采集的数据用于机器人和汽车应用。
  • 根据人的姿势对人的动作进行分类。
  • 估计人、动物和物体上的关键点。
  • 只需 30 分钟的记录数据就可以创建自定义声音。

开始使用

  • 在 TAO 工具套件入门页面观看教学视频和快速入门指南。详情请见:https://developer.nvidia.com/...
  • 从 NGC 下载计算机视觉或对话式 AI 模型。
  • 观看新功能的演示视频,包括 Rest API、TensorBoard 和预训练模型。详情请见:https://www.nvidia.com/en-us/...[]=TAO&facet.mimetype[]=event%20session&headerText=All%20Sessions&layout=list&page=1&q=-&sort=date&sortDir=desc

与 TAO 工具套件相关的其他资源和用例

  • 使用 NVIDIA Isaac Sim 和 NVIDIA TAO 开发和部署 AI 机器人。
  • 使用 TAO 工具套件检测动物健康和表型
  • RocketBoots 提高自动化员工队伍管理的推理性能
  • 改变零售行为识别用例的开发工作

如需下载 TAO Toolkit 5.22. 详见链接:
https://catalog.ngc.nvidia.co...

推荐阅读
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息