下载
中文
注册

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

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

  1. 参见配置文件样例描述文件参数说明构造单算子描述文件。本章节以构造format为ND的Add单算子为例进行说明。
  2. 以CANN软件包运行用户,将1构造的单算子描述文件上传到开发环境任意目录,例如$HOME/singleop/目录下。
  3. 执行如下命令生成离线模型。(如下命令中使用的目录以及文件均为样例,请以实际为准)
    atc --singleop=$HOME/singleop/add.json --output=$HOME/singleop/out/op_model --soc_version=<soc_version>

    关于参数的详细解释以及使用方法请参见--singleop

  4. 若提示如下信息,则说明模型转换成功。
    ATC run success

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