tao-train-single-step

Automate training, evaluation, and export for TAO models.

83|20|Updated Apr 16, 2026
One-click install
npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-train-single-step
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: tao-train-single-step
Source: https://github.com/NVIDIA-TAO/tao-skill-bank/tree/main/skills/applications/tao-train-single-step
Command: npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-train-single-step

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires docker, nvidia-container-toolkit, and includes scripts (resource) and references (resource) components.

What problem does it solve?

Simplifies the process of training, evaluating, and exporting any TAO model in a single workflow.

Core Features & Use Cases

  • Single Workflow: Perform training, evaluation, and export in one continuous process.
  • Versatile Model Support: Compatible with various TAO models like clip, nvdinov2, grounding_dino.
  • Use Case: Quickly fine-tune a pre-trained TAO model on a new dataset, evaluate its performance, and export the model for deployment.

Quick Start

To initiate a training, evaluation, and export workflow for a TAO model, use the tao-train-single-step skill with the required parameters, such as model type, dataset URI, and platform.

Frequently Asked Questions about tao-train-single-step

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I fine-tune, evaluate, and export a TAO model in a single workflow?

You can fine-tune, evaluate, and export a TAO model in a single workflow by automating the process for pretrained models, streamlining iterative training and evaluation without needing AutoML or DEFT loops.

Does TAO model training require Docker and NVIDIA-container-toolkit?

Yes, TAO model training requires Docker and NVIDIA-container-toolkit for execution. These dependencies are necessary to run the containerized environment for the single-step training workflow.

Which TAO models are compatible with single-step training and export?

Single-step training and export supports various TAO models, including clip, nvdinov2, and grounding_dino. It is designed to be versatile for fine-tuning different pretrained model architectures on new datasets.

What is the best way to iteratively train and evaluate TAO models without AutoML?

The best way to iteratively train and evaluate TAO models without AutoML is using a single-step workflow that automates fine-tuning and performance evaluation on a new dataset, culminating in model export for deployment.

Can I use this single-step TAO training workflow with AutoML or DEFT loops?

No, this single-step TAO training workflow is specifically designed for iterative training and evaluation without AutoML or DEFT loops. It simplifies the process by performing training, evaluation, and export continuously.