GE图引擎ATC工具--json参数详解
2026/9/10 1:49:51 网站建设 项目流程

--json

【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

全量芯片支持

IPV350:不支持

功能说明

离线模型、原始模型文件、GE dump图结构文件转换为JSON文件的路径和文件名。

如果是已有的离线模型转换为JSON文件,在转换后的文件中还可以查看原始模型转换为该离线模型时,使用的基础版本号(比如ATC软件版本信息,OPP算子包版本信息等)以及当时模型转换使用的atc命令。

关联参数

  • 离线模型转换为JSON文件

    该参数需要与--mode=1、--om参数配合使用。

  • 原始模型文件转换为JSON文件

    该参数需要与--mode=1、--om参数、--framework配合使用。

    原始模型为MindSpore框架时,即--framework=1时,不支持转换为JSON文件。

  • GE dump图结构文件转JSON文件

    该参数需要与--mode=5、--om参数配合使用。

    仅支持dump出的ge_proto*.txt格式文件转成JSON文件。

参数取值

参数值:JSON文件的路径和文件名。

参数值格式:路径和文件名:支持大小写字母(a-z,A-Z)、数字(0-9)、下划线(_)、短横线(-)、句点(.)、中文汉字。

推荐配置及收益

无。

示例

  • 离线模型转换为JSON文件

    atc --mode=1 --om=$HOME/module/out/tf_resnet50.om --json=$HOME/module/out/tf_resnet50.json ...
  • 原始模型文件转换为JSON文件

    atc --mode=1 --om=$HOME/module/resnet50_tensorflow.pb --json=$HOME/module/out/tf_resnet50.json --framework=3 ...
  • GE dump图结构文件转JSON文件

    atc --mode=5 --om=$HOME/module/ge_proto_00000000_PreRunBegin.txt --json=$HOME/module/out/ge_proto.json ...

依赖约束

无。

【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询