| 名称 | tao-train-image-classification |
| 描述 | 基于PyTorch的TAO图像分类（Classification PyT）。支持包括FAN、EfficientNet、ResNet在内的多种骨干网络，并提供蒸馏、量化能力以便部署。本技能用于训练、评估、蒸馏、量化、导出及推断TAO图像分类模型。数据集类型为image_classification，格式为classification_pyt，监控指标为val_acc_1。技能自动启用AutoML以优化训练超参，也支持关闭AutoML执行手动训练。支持多GPU、多节点训练。更多细节见英文原文档。 |
TAO图像分类（PyTorch）Skill tao-train-image-classification
该技能用于基于NVIDIA TAO框架的图像分类模型训练与部署。支持多种骨干网络（FAN/EfficientNet/ResNet等）、AutoML超参优化、知识蒸馏、量化、TensorRT导出和推理。关键词：TAO、图像分类、PyTorch、AutoML、模型训练、推理、量化、TensorRT、视觉AI。