domains / families / science / tao-train-image-classification
tao-train-image-classification
separate
PyTorch-based TAO image classification. Supports a wide range of backbones (FAN, EfficientNet, ResNet, etc.)
When to use
- train image classifier
- TAO classification
- ResNet/EfficientNet/FAN backbone classifier
- classification-pyt
- distilling
- quantizing
- exporting
- running inference
When not to use
- non-train actions such as evaluate, inference, export, and deploy flows stay in this model skill
Skill page
Source
| Repository | Path |
|---|
| NVIDIA/skills | skills/tao-train-image-classification/SKILL.md |