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.