--om
功能说明
离线模型(.om)、原始模型文件(例如Caffe框架的.prototxt,TensorFlow框架的.pb等)、GE dump图结构文件(.txt)的路径和文件名。
关联参数
参数取值
参数值:离线模型(.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
- 离线模型转换为json:
- 若--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推理产品
依赖约束
无。
父主题: 输入选项