--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),仅供参考