一键式精度分析工具,提供了对训练网络进行精度分析的常用功能,主要包括:
该工具封装了TF Adapter的运行参数,帮助用户更方便的使能相关业务特性,同时对CANN Toolkit包中的精度分析工具进行了封装和功能扩展,从而帮助开发人员快速分析精度问题。
├── resnet // 训练工作目录
│ ├── __init__.py
│ ├── imagenet_main.py // 基于Imagenet数据集训练网络模型
│ ├── imagenet_preprocessing.py // Imagenet数据集数据预处理模块
│ ├── resnet_model.py // resnet模型文件
│ ├── resnet_run_loop.py // 数据输入处理与运行循环(训练、验证、测试)
│ ├── cifar10_main // 训练执行入口文件
│ ├── ...
│ ├── precision_tool // 一键式精度分析工具目录
│ │ ├── cli.py
│ │ ├── ...
CANN运行环境(包含昇腾AI处理器,即启动NPU训练的环境),在训练精度调优工作中,主要用于训练时精度数据采集;
CANN开发环境(即安装CANN Toolkit软件的环境),在训练精度调优工作中,主要用于精度数据分析。