Lightlines
By nvidia

tao-run-automl

nvidia1.6k installs

Find better settings for your NVIDIA TAO models.

What it does

Tunes NVIDIA TAO model actions through automated hyperparameter searches and returns the winning configuration, model artifact, and metric comparisons.

When to use it

Use for explicit TAO AutoML or hyperparameter optimization requests, including tuning training, evaluation, inference, distillation, pruning, or quantization.

How to use it

Give it a TAO tuning request and the selected action's data or checkpoints; it produces experiment artifacts and identifies the winning configuration and model artifact.

What you provide

  • Your files

Uses

Good to know

  • acts without asking

Latest review

What you need · 10

The compatibility declaration requires Docker for container-backed execution.

The compatibility declaration requires NVIDIA Container Toolkit.


Access · 2

This skill

nvidia-tao-automl

Execute

Installs the AutoML platform package

trial checkpoints

Write

Deletes unneeded trial checkpoints