昇腾社区首页
中文
注册

如何将算子描述文件转成离线模型

本节给出单算子描述文件转成离线模型的详细步骤。

  1. 参见什么是单算子描述文件中的参数解释以及配置文件样例构造单算子描述文件。本章节以构造format为ND的Add单算子add.json为例进行说明。
  2. 以CANN软件包运行用户,将步骤1构造的单算子描述文件上传到开发环境任意目录,例如$HOME/singleop/目录下。
  3. 执行如下命令生成离线模型。(如下命令中使用的目录以及文件均为样例,请以实际为准)
    atc --singleop=$HOME/singleop/add.json --output=$HOME/singleop/out/op_model --soc_version=<soc_version>
    • --singleop:用于指定add.json单算子描述文件。
    • --output:转换后的离线模型存放路径。
    • --soc_version:昇腾AI处理器的型号。

    关于参数的详细解释请参见参数说明

  4. 若提示如下信息,则说明模型转换成功。若模型转换失败,请参见错误码参考章节进行辅助定位。
    1
    ATC run success, welcome to the next use.
    

    成功执行命令后,在output参数指定的路径下,可查看离线模型文件*.om。