通过 NVIDIA TAO 5.5 实现全新基础模型和增强训练功能

文章预览

NVIDIA TAO 是一个用于简化和加速 AI 模型开发与部署的框架。有了它，您无需具备深厚的 AI 专业知识，就可以使用预训练模型、利用自己的数据对这些模型进行微调，并针对特定用例优化这些模型。 TAO 能够与 NVIDIA 软硬件生态系统无缝集成，提供实现高效 AI 模型训练、部署和推理的工具，加快 AI 驱动型应用的上市时间。 图 1.  NVIDIA TAO 架构 图 1 显示 TAO 支持 PyTorch、TensorFlow 和 ONNX 等框架。可以在多个平台上完成训练，生成的模型可以部署在各种推理平台上，例如 GPU、CPU、MCU 和 DLA。 NVIDIA 最近发布的 TAO 5.5 带来了最先进的基础模型和开创性的功能，加强了一切 AI 模型的开发。新功能包括： 多模态传感器融合模型 ：将来自多个传感器的数据整合到统一的鸟瞰图（BEV）表现方式中，保留了几何和语义信息。 文本提示自动标记 ：使用文本提示来自动 ………………………………