--om

功能说明

离线模型（.om）、原始模型文件（例如Caffe框架的.prototxt，TensorFlow框架的.pb等）、GE dump图结构文件（.txt）的路径和文件名。

关联参数

• 若--mode取值为1：
  • 离线模型转换为json

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

  • 原始模型文件转换为json

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

• 若--mode取值为5：

  GE dump图结构文件转json，该参数需要与--mode=5、--json参数配合使用。

• 若--mode取值为6：

  则只需要与--mode参数配合使用。

参数取值

参数值：离线模型（.om）、原始模型文件（例如Caffe框架的.prototxt，TensorFlow框架的.pb）或GE dump图结构文件（.txt）的路径。

参数值格式：路径和文件名：支持大小写字母（a-z，A-Z）、数字（0-9）、下划线（_）、中划线（-）、句点（.）、中文字符。

推荐配置及收益

无。

示例

• 若--mode取值为1
  • 离线模型转换为json：

    --mode=1 --om=$HOME/module/out/tf_resnet50.om  --json=$HOME/module/out/tf_resnet50.json

    在转换后的json文件中，可以查看原始模型转换为该离线模型时，使用的基础版本号，以及当时转换使用的atc命令，示例如下：（如下示例中的版本号都为样例，请以实际查询的为准）

    说明：离线模型转换为json文件可以查看基础版本号，必须保证使用的软件为1.77.22.6.220及之后的版本，之前版本无法查阅该信息；atc_cmdline参数必须为1.78.23.34.230及之后的软件版本方可查阅。

       {
          "key": "opp_version",
          "value": {
            "s": "1.78.23.34.230"
          }
        },
        ...
        {
          "key": "atc_version",
          "value": {
            "s": "1.78.23.34.230
          }
        },
        ...
    
        { 
           "key": "atc_cmdline",
           "value": {
             "s": "xxx/atc.bin --model ./resnet50_tensorflow*.pb  --framework 3 --output ./out/tf_resnet50 --soc_version Ascend310"
           }
         },
        ...
        {
          "key": "soc_version",
          "value": {
            "s": "Ascend310"
           }
         },
        ...

  • 原始模型文件转换为json

    --mode=1 --om=$HOME/module/resnet50_tensorflow*.pb  --json=$HOME/module/out/tf_resnet50.json  --framework=3

• 若--mode取值为5
  GE dump图结构文件转json：

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

• 若--mode取值为6

  atc --mode=6 --om=$HOME/module/out/tf_resnet50.om

支持的芯片型号

昇腾310 AI处理器

昇腾310P AI处理器

昇腾910 AI处理器

依赖约束

无。