昇腾社区首页
中文
注册
开发者
下载

比对步骤(PyTorch)

保存基于GPU执行模型训练生成的数据文件作为标准算子的训练数据文件,将保存基于昇腾AI处理器执行模型训练生成的数据文件作为待比对算子的训练数据文件,执行Vector比对。

Vector比对命令行方式操作步骤:

  1. 以HwHiAiUser用户登录操作系统。
  2. 进入/home/HwHiAiUser/Ascend/ascend-toolkit/latest/toolkit/tools/operator_cmp/compare目录。
  3. 执行Vector比对命令,样例命令如下:

    python3 msaccucmp.py compare -m /home/Jason.h5 -g /home/resnet50_dump.h5 -out /home/HwHiAiUser/result

    Vector比对结果result_*.csv文件内容如图1所示。
    图1 模型比对结果

    比对PyTorch算子的训练结果时,仅支持-m、-g、-out、-alg和-p五个参数。比对结果各列参数说明如表1

PyTorch比对的结果中若有IR算子存在精度问题,则需要记录存在精度问题的IR算子名,并根据《PyTorch网络模型移植&训练指南》中的“精度调测>调测过程>精度调优方法>单算子溢出检测>IR与TBE算子映射”章节将IR算子转换为TBE算子后进行精度问题分析。