--om

功能说明

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

关联参数

若--mode取值为1：
- 离线模型转换为json
  --om需要与--mode=1、--json参数配合使用。
- 原始模型文件转换为json
  --om需要与--mode=1、--json、--framework参数配合使用。
若--mode取值为5：
GE dump图结构文件转json，--om需要与--mode=5、--json参数配合使用。
若--mode取值为6：
针对已有的离线模型，显示模型信息等信息，则--om只需要与--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命令，示例如下：

   {
      "key": "opp_version",
      "value": {
        "s": "<version>"
      }
    },
    ...
    {
      "key": "atc_version",
      "value": {
        "s": "<version>"
      }
    },
    ...

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

原始模型文件转换为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

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/500 A2推理产品

Atlas A2训练系列产品/Atlas 800I A2推理产品

依赖约束

无。

父主题： 输入选项