tao-run-automl-deft-pipeline

Automate the end-to-end NVIDIA TAO training pipeline with AutoML and DEFT.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

This Skill automates the end-to-end training pipeline for NVIDIA TAO models, including AutoML baseline, DEFT loop, and AutoML refinement.

Core Features & Use Cases

  • End-to-End Training: Automates the three-phase training pipeline for NVIDIA TAO models.
  • AutoML Baseline: Performs Hyperparameter Optimization (HPO) on the initial dataset.
  • DEFT Loop: Implements an iterative data-improvement loop for the DEFT application.
  • AutoML Refinement: Refines the model on the augmented dataset after the DEFT loop.
  • Use Case: Ideal for users who need to run an end-to-end training pipeline for AOI inspection and other DEFT applications on NVIDIA TAO models.

Quick Start

Run the AOI training pipeline with the tao-run-automl-deft-pipeline skill.

Frequently Asked Questions about tao-run-automl-deft-pipeline

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

FAQPage Schema
How do I automate the end-to-end training pipeline for NVIDIA TAO models?

Automating the end-to-end NVIDIA TAO training pipeline requires executing a three-phase workflow: AutoML baseline for hyperparameter optimization, a DEFT loop for iterative data improvement, and AutoML refinement on the augmented dataset. This pipeline automates these phases sequentially for DEFT applications like AOI inspection.

What is the DEFT loop in an automated training pipeline?

The DEFT loop is an iterative data-improvement process applied after an initial AutoML baseline. It enhances the dataset quality before running final AutoML refinement, ensuring the NVIDIA TAO model is trained on optimized data for tasks such as AOI inspection.

Do I need Docker and nvidia-container-toolkit to run NVIDIA TAO AutoML workflows?

Yes, Docker and nvidia-container-toolkit are required dependencies for execution. You must install and configure these container tools in your environment to run the end-to-end training pipeline for NVIDIA TAO models successfully.

Can I use this automated pipeline for AOI inspection model training?

Yes, the automated pipeline is ideal for AOI inspection and other DEFT applications. It applies the AutoML baseline, DEFT loop, and AutoML refinement phases directly to end-to-end training workflows for NVIDIA TAO models.

What is the best way to run hyperparameter optimization for NVIDIA TAO models?

Running hyperparameter optimization (HPO) for NVIDIA TAO models is best handled by the AutoML baseline phase. It initializes the end-to-end pipeline by optimizing model parameters on the initial dataset before proceeding to the DEFT data-improvement loop.