全网层信息文件
以下介绍通过ATC模型转换工具获取离线模型的操作步骤,更多操作请参见《ATC工具使用指南》。
- 登录安装了Ascend-cann-toolkit开发套件包的昇腾AI环境。
- 获取原始模型文件并保存在任意目录下。
例如:resnet50*.onnx
- 执行ATC模型转换。
atc --model=$HOME/module/resnet50*.onnx --framework=5 --output=$HOME/module/out/onnx_resnet50 --soc_version=<soc_version>
若提示如下信息,则说明模型转换成功。1
ATC run success
成功执行命令后,在--output参数指定的路径下,可查看离线模型(如:onnx_resnet50.om)。
- 生成json文件。
atc --mode=1 --om=$HOME/module/out/onnx_resnet50.om --json=$HOME/module/out/onnx_resnet50.json
若提示如下信息,则说明转换json文件成功。
1
ATC run success
成功执行命令后,在--json参数指定的路径下可查看转换后的json文件。
父主题: GPU vs NPU(ONNX推理)