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