数据与分析 ✓ NVIDIA · 官方 automlhyperparameter-optimizationnvidia-taowandbbayesian-searchhyperbandmodel-training

TAO AutoML 调参技能

为 NVIDIA TAO 网络运行可审查的自动超参数优化流程。

FollowSkills 评估 · FSRS-2.0
不推荐
54/ 100 五分制 2.7 / 5
信任安全17 / 25 · 3.4/5

文档要求平台、凭据、数据、镜像、资源和用户确认后才启动,并禁止询问或打印秘密;但允许 Bash/Write,涉及 pip 安装、外部服务、S3/WandB/LLM,且示例中的 curl 可能将 API 密钥放入命令行,缺少更严格的权限隔离、回滚和敏感数据流控制,因此扣分。

可靠稳定8 / 20 · 2.0/5

预检、模型支持门、基线评估、失败分类、恢复和状态查询描述较完整;但 SKILL.md 与引用文件存在接口和路径不一致(如 platform_sdk 与 sdk、构造参数及工作流路径),依赖模型技能和未提供文件,静态上无法复现关键路径,故不超过静态上限并扣分。

适用触发10 / 15 · 3.3/5

目标用户、TAO AutoML/HPO 场景、平台选择、算法和非适用条件基本明确;但触发边界仍依赖读取其他模型与平台技能,未提供中文交互支持,LLM 功能可能依赖海外 NVIDIA 端点,面向中国大陆的可达性未说明,因此扣分。

规范维护9 / 15 · 3.0/5

有 front matter、Apache-2.0、版本号、作者、标签、引用地图、分层参考文档、示例和常见故障;但作者格式被基准报告指出不符合规范,缺少明确维护责任、变更日志、更新路径,且引用内容存在过时或冲突接口,故扣分。

有效结果6 / 15 · 2.0/5

工作流覆盖从预检、搜索空间、基线评估、运行监控到结果交接,理论上能直接支持 AutoML 运行;基准报告显示一次任务取得较高正确性和有效性结果,但只有单任务且无本次执行,模型特定依赖和接口冲突限制了可验证的直接可用性,故按静态上限保守扣分。

证据核验4 / 10 · 2.0/5

存在固定修订版本、具体命令、返回结构、示例和 NVSkills-Eval 报告,提供了一定审计线索;但报告仅含单任务,未见覆盖关键路径的真实 CI 与提交测试套件,无法静态独立复现,多数结论仍依赖未提供的模型、平台和脚本文件,因此扣分。

证据充分度: 评估于 2026年7月20日 审查版本 55f18499943e
上游仓库在本次评估后已有新提交;当前评分仍对应所示审查版本,可能尚未覆盖最新改动。
使用前请注意
  • 启动真实训练任务前,核对 AutoMLRunner 构造参数、平台 SDK 接口和文档路径的冲突。
  • 不要在 shell 命令、日志或配置中暴露 LLM、WandB、云平台或 Hugging Face 凭据;尤其避免直接复制带 API 密钥的 curl 示例。
  • 确认 NVIDIA LLM 端点及 S3/WandB 等外部服务在目标网络(包括中国大陆)可达,并准备不可用时的替代方案。
  • 该评估只覆盖 tao-run-automl/SKILL.md 及其随附材料;实际运行仍必须验证对应模型和平台技能、schema、镜像与依赖。
查看完整评分方法 →

这个 Skill 能做什么,适合哪些场景?

该技能使用 AutoMLRunner 为 NVIDIA TAO 网络执行自动机器学习和超参数优化。它支持贝叶斯搜索、Hyperband、ASHA、BOHB、LLM 引导搜索、混合搜索和 autoresearch,并可在 Brev、SLURM、Kubernetes 或 Docker 平台上运行。流程会先检查模型、平台、数据、凭据、容器镜像和计算资源,再运行基线评估并请求启动确认。它还支持 WandB 跟踪、结果解释、自定义评估钩子、断点恢复和最终模型评估。

读取选定模型与平台技能及其元数据、训练 schema 和模板;执行平台、依赖、数据可见性、凭据、容器和资源预检;验证 nvidia-tao-automl 是否可导入;构建嵌套训练配置和搜索空间;在推荐任务前运行自动基线评估;调用 AutoMLRunner 生成推荐、提交训练任务、提取指标并反馈给优化器;通过平台 SDK 查询任务状态、日志和失败分析;记录推荐指标、失败原因、最佳检查点、结果路径和最终评估结果。

  1. 需要为受支持的 TAO 视觉网络在小到中等预算下选择训练超参数的团队,可使用默认贝叶斯搜索。
  2. 需要并行尝试大量配置且训练支持早期停止的用户,可使用 Hyperband 或 ASHA。
  3. 希望在 Brev、SLURM、Kubernetes 或本地 Docker 上执行 TAO AutoML 的用户,可按所选平台完成预检和任务提交。
  4. 需要用任务验证指标而非训练损失比较模型的团队,可配置评估函数和最终评估回调。
  5. 希望跟踪实验、恢复已有工作区或检查失败推荐的用户,可使用 WandB、监控和恢复流程。

这个 Skill 有哪些优点和局限?

优点
  • 覆盖多种超参数优化算法和 TAO SDK 平台。
  • 要求在提交推荐任务前完成模型支持、数据、凭据、镜像和资源检查。
  • 支持基线评估、每次推荐评估、最终评估、监控、失败分类和结果交接。
  • 可使用模型技能中的训练 schema 约束搜索空间,降低无效配置风险。
局限
  • 不能脱离模型技能的支持门禁自行推断搜索空间;缺少必需 schema 或评估能力时会停止。
  • 运行依赖 Docker、nvidia-container-toolkit、Python 和 nvidia-tao-automl;部分流程还可能需要对象存储或平台 CLI。
  • AutoML 任务可能耗时很长,具体运行时间和成本取决于预算、数据、平台和资源;源材料没有定价信息。
  • 源材料未提供该技能自身的测试套件或已验证的平台覆盖证据。

如何安装这个 Skill?

按仓库 README 使用:npx skills add nvidia/skills --skill tao-run-automl --yes。该命令安装指定技能;源材料未说明具体安装目录。运行工作流还需要 Docker、nvidia-container-toolkit、Python 以及 nvidia-tao-automl 包。

如何使用这个 Skill?

安装后,向兼容的 Agent Skills 客户端提出类似“为我的 NVIDIA TAO 网络运行 AutoML 超参数优化,使用 bayesian 算法和指定平台”的请求,并提供模型、平台、训练/评估数据、结果目录、GPU/节点数量、镜像、指标和预算。技能会在启动前展示评审信息并请求确认;源材料未提供完整的用户项目配置示例。

这个 Skill 与同类方案有什么区别?

技能将 bayesian 定位为小到中等预算的默认选择;Hyperband/ASHA 适合可早停的多配置搜索;BOHB/DEHB 面向多保真搜索;PBT 面向需要训练期间变异的长任务;llm、hybrid 和 autoresearch 需要用户明确要求并配置端点。

常见问题

它会在没有确认的情况下提交训练任务吗?
不会。每次 AutoML 运行都必须先展示启动评审并获得用户确认。
它是否只支持某一种 TAO 运行平台?
不是。技能列出 Brev、SLURM、Kubernetes 和 Docker,并要求先执行所选平台的预检。
缺少 nvidia-tao-automl 包时会怎样?
技能会显示由 versions.yaml 解析出的安装命令,并在安装前询问用户;不会自行臆造依赖版本。
如果所有推荐任务都失败,会继续增加试验次数吗?
不会。它应总结共同根因并停止,而不是静默追加无效预算。

同仓库的其他 Skills

均来自 NVIDIA/skills

数据与分析 ✓ NVIDIA · 官方

TAO 标准训练流程

为 TAO 模型执行标准的训练、评估与导出流程。

数据与分析 ✓ NVIDIA · 官方

TAO 人员重识别训练技能

训练和部署用于跨摄像头人员匹配的 TAO ReID 模型。

数据与分析 ✓ NVIDIA · 官方

TAO RT-DETR 目标检测

指导使用 NVIDIA TAO 训练并部署实时 RT-DETR 检测模型。

数据与分析 ✓ NVIDIA · 官方

TAO Grounding DINO 开放词汇检测

用文本提示训练和部署无需固定类别词表的目标检测模型。

数据与分析 ✓ NVIDIA · 官方

TAO OneFormer 通用分割技能

用单一任务条件化模型训练并部署全景、实例和语义图像分割。

数据与分析 ✓ NVIDIA · 官方

TAO SegFormer 语义分割技能

为 TAO SegFormer 提供语义分割训练、评估与部署指导。

数据与分析 ✓ NVIDIA · 官方

TAO AutoML + DEFT 三阶段训练管线

为 AOI 与其他 DEFT 应用串联超参数优化、数据改进和最终模型精调。

数据与分析 ✓ NVIDIA · 官方

Mask Grounding DINO 训练与部署技能

用文本提示训练、评估并部署开放词汇实例分割模型。

数据与分析 ✓ NVIDIA · 官方

TAO MAE 视觉预训练技能

指导 NVIDIA TAO 使用掩码自编码器进行视觉预训练与微调。

数据与分析 ✓ NVIDIA · 官方

TAO Deformable DETR 目标检测

帮助 AI 代理训练、评估、导出、量化并部署轻量级 Deformable DETR 目标检测模型。

开发与工程 ✓ NVIDIA · 官方

TAO Sparse4D 训练技能

帮助代理训练、评估、导出、量化和推理多摄像头时序3D检测跟踪模型。

数据与分析 ✓ NVIDIA · 官方

OCDNet 场景文本检测

用 TAO Toolkit 训练和部署任意方向场景文本检测模型。

数据与分析 ✓ NVIDIA · 官方

TAO DEFT AOI 检测优化循环

为 PCB 视觉检测模型自动执行评估、缺陷增强、挖掘、重训与部署门控。

开发与工程 ✓ NVIDIA · 官方

TAO 技能库能力发现

通过打包的清单快速查询 TAO Skill Bank 支持的工作流、模型、平台和 AutoML 能力。

数据与分析 ✓ NVIDIA · 官方

TAO PyT 图像分类

用 NVIDIA TAO 训练、评估并部署 PyTorch 图像分类模型。

数据与分析 ✓ NVIDIA · 官方

TAO ChangeNet 根因分析

基于图像证据定位 NVIDIA TAO Visual ChangeNet 分类实验的模型失败原因。

数据与分析 ✓ NVIDIA · 官方

TAO DAFT 数据集验证器

用 tao-daft validate 检查 TAO DAFT 数据集的结构、架构与交叉引用错误。

数据与分析 ✓ NVIDIA · 官方

TAO DINO 目标检测

指导使用 NVIDIA TAO 训练、评估和部署 DINO 目标检测器。

开发与工程 ✓ NVIDIA · 官方

TAO HuggingFace 模型集成助手

将 HuggingFace 计算机视觉模型接入 NVIDIA TAO 的训练、导出与 TensorRT 部署流程。

数据与分析 ✓ NVIDIA · 官方

TAO Depth Anything v2 深度估计

使用 NVIDIA TAO 从单张 RGB 图像训练、评估、导出和推理单目深度模型。

相关 Skills