下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

--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命令,示例如下:

       1
       2
       3
       4
       5
       6
       7
       8
       9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      24
      25
      26
      27
      28
      29
         {
            "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 推理系列产品(Ascend 310P处理器)

Atlas 训练系列产品

Atlas 200/500 A2推理产品

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

依赖约束

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词