About

Imported from nvidia/skills (skills/tao-train-single-step/SKILL.md) via skills.sh. Install upstream with npx skills add nvidia/skills --skill tao-train-single-step. Copyright stays with the author (Apache-2.0).

Normal Train

Name: tao-train-single-step
Author: nvidia

Standalone install? If this session was not initialized by the TAO skill bank plugin, run the tao-setup skill first (host preflight, credentials, cross-skill discovery).

Standard supervised fine-tuning: train a model on a labeled dataset, optionally evaluate, then optionally export. The most common TAO workflow for adapting a pretrained model to a new dataset.

Steps

train — executed through AutoML when the selected model has automl_enabled: true and automl_policy is on; set automl_policy=off for a plain single training run
eval — executed if eval_dataset_uri is resolved
export — optional, on user request after training

Prerequisites

The selected model skill's resolved container_image is the default training runtime. Do not replace it with a host venv, uv environment, generic training image, or hand-written trainer unless the user explicitly requests that execution mode. SDK/controller Python environments are control-plane-only; the model action remains container-backed.

Required

model: A compatible TAO model (e.g., clip, nvdinov2, grounding_dino)
train_dataset_uri: URI of the training dataset (e.g., s3://bucket/train/)
platform: Ask from the generated supported-platform list: ${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_platforms.py --format text
container image confirmation: resolve the default image from the selected model/action config, show it to the user, and require confirmation or image=<override> before creating runner files or submitting training.

Optional

eval_dataset_uri: Some model skills mark this as required — check the resolved model skill before treating it as optional.
base_checkpoint: If not provided, defaults to the NGC pretrained checkpoint listed in the model skill, or trains from scratch if no NGC checkpoint exists.
automl_policy: on by default; set off to bypass model-level AutoML for this run while leaving model metadata unchanged. Use only on / off in new launch settings.
image override: Use image=<override> to pin a specific TAO toolkit build after reviewing the resolved default.

Launch Intake

After the user confirms they want this standard train/eval/export workflow, ask which supported platform they intend to run on. Generate the choices with scripts/list_tao_platforms.py --format text; do not scan platform docs or folders.

Before creating a plain train runner, inspect the selected model's metadata with scripts/list_tao_models.py --scope automl --format json or read skills/models/<network>/references/skill_info.yaml. If automl_enabled is true and the helper reports a valid train schema for that model, route the train stage through skills/applications/tao-run-automl by default. Only stay on the plain train path when automl_policy=off, the user explicitly asks for no HPO/AutoML, or AutoML is enabled but not runnable because the model's train schema is not packaged yet.

Also ask whether long-running monitoring should stay enabled and how many minutes between status updates. Defaults: enabled, 5 minutes.

After the model/action are known, run scripts/resolve_tao_image.py --model <network> --action train --format text and ask whether to use the resolved image or an image=<override>. Do not create the tao-train-single-step runner until the image is confirmed.

After platform selection, run scripts/list_tao_platforms.py --platform <platform> --format text and ask only for credentials relevant to that platform, plus any selected-model credentials. Do not ask for unrelated platform credentials.

Use it

Copy one of these into your project. Installing also returns the manifest and these snippets.

yaml

targets:
  - https://api.opensmartroute.ai/api/v1/registry/nvidia-skills-tao-train-single-step/manifest   # or paste the manifest below

Manifest

An Open Capability Manifest: the router reads it to know what this does, what it costs and when to pick it.

nvidia-skills-tao-train-single-step.ocm.jsonjson

{
  "ocm": "1",
  "id": "nvidia-skills-tao-train-single-step",
  "kind": "skill",
  "name": "tao-train-single-step",
  "description": "Standard single-step train/eval/export workflow for any TAO model. Use when training a TAO model on a dataset without iterative data augmentation, AutoML, or DEFT loops. Trigger phrases include \"single train run\", \"train then evaluate then export\", \"plain TAO training\", \"normal training\", \"no AutoML\", \"skip the loop\". Routes through the per-model SKILL.md for action specifics and through `tao-launch-workflow` for platform/credentials/dataset intake.",
  "publisher": "nvidia",
  "version": "1.0.0",
  "capabilities": {
    "domains": [
      "data_analysis"
    ],
    "tags": [
      "skill-md",
      "training",
      "single-step",
      "generic",
      "skills-sh"
    ],
    "languages": [
      "en"
    ]
  },
  "quality_prior": 0.6,
  "examples": [
    "Standard single-step train/eval/export workflow for any TAO model. Use when training a TAO model on a dataset without iterative data augmentation, AutoML, or DEFT loops. Trigger phrases include \"single train run\", \"train then evaluate then export\", \"plain TAO training\", \"normal training\", \"no AutoML\", \"skip the loop\". Routes through the per-model SKILL.md for action specifics and through `tao-launch-workflow` for platform/credentials/dataset intake."
  ],
  "primary": false,
  "metadata": {
    "source": {
      "provider": "skills.sh",
      "repository": "https://github.com/nvidia/skills",
      "path": "skills/tao-train-single-step/SKILL.md",
      "ref": "HEAD",
      "url": "https://www.skills.sh/nvidia/skills/tao-train-single-step",
      "key": "nvidia/skills/skills/tao-train-single-step/SKILL.md"
    },
    "compatibility": "Requires docker + nvidia-container-toolkit. Workflows declare additional requirements.",
    "allowed_tools": [
      "Read",
      "Bash",
      "Write"
    ],
    "license": "Apache-2.0"
  },
  "instructions": "# Normal Train\n\n> **Standalone install?** If this session was not initialized by the TAO skill bank plugin, run the `tao-setup` skill first (host preflight, credentials, cross-skill discovery).\n\nStandard supervised fine-tuning: train a model on a labeled dataset, optionally evaluate, then optionally export. The most common TAO workflow for adapting a pretrained model to a new dataset.\n\n## Steps\n\n1. **train** — executed through AutoML when the selected model has\n   `automl_enabled: true` and `automl_policy` is `on`; set\n   `automl_policy=off` for a plain single training run\n2. **eval** — execut",
  "cost": {
    "context_tokens": 919
  }
}

Fetch it by URL: GET /api/v1/registry/nvidia-skills-tao-train-single-step/manifest?version=1.0.0

Reviews

Star ratings from people who tried it. One review per account; edit yours any time.

No reviews yet. Install it, try it, and be the first to rate it.