如何将算子描述文件转成离线模型
本节给出单算子描述文件转成离线模型的详细步骤。
- 参见配置文件样例和描述文件参数说明构造单算子描述文件。本章节以构造format为ND的Add单算子为例进行说明。
- 以CANN软件包运行用户,将1构造的单算子描述文件上传到开发环境任意目录,例如$HOME/singleop/目录下。
- 执行如下命令生成离线模型。(如下命令中使用的目录以及文件均为样例,请以实际为准)
atc --singleop=$HOME/singleop/add.json --output=$HOME/singleop/out/op_model --soc_version=<soc_version>
关于参数的详细解释以及使用方法请参见--singleop。
- 若提示如下信息,则说明模型转换成功。
ATC run success
成功执行命令后,在output参数指定的路径下,可查看离线模型文件*.om。
父主题: 单算子模型转换