Train image classifiers ready for deployment.
What it does
Trains and prepares TAO image classifiers for evaluation, inference, and deployment.
When to use it
Use for training, evaluating, distilling, quantizing, exporting, or running inference with a TAO PyTorch image-classification model.
How to use it
Give it image datasets and a TAO classification training request; it saves training results in an output folder.
What you provide
- Your files
Uses
Latest review
What you need · 5
Requires Docker for the container-based workflow.
Requires NVIDIA Container Toolkit alongside Docker.
Requires at least one GPU with 16GB or more VRAM per GPU; two GPUs are recommended.
Access · 1
This skill
S3
Read
Downloads image datasets from S3