tao-train-image-classification - 训练与部署 TAO PyTorch 图像分类模型
训练、评估、蒸馏、量化、导出并推理基于 PyTorch 的 TAO 图像分类模型。
标签
更新于: 2026-09-28该技能可以做什么
- 训练图像分类模型
- 评估模型性能
- 蒸馏分类模型
- 量化模型
- 导出训练模型
- 运行模型推理
输入
- 训练图像目录
- 验证图像目录
- 测试图像目录
- 类别标签文件
- 骨干网络权重
- 预训练模型检查点
- 动作配置规范
- 可写结果目录
输出
- 训练模型检查点
- 评估指标
- 量化模型文件
- 导出模型文件
- TensorRT 引擎
- 推理结果
要求
- Docker
- NVIDIA Container Toolkit
- 独立运行时的 TAO 技能库初始化
- PyTorch 模型支持