将ONNX模型转换为OM模型
将导出ONNX模型导出的“resnet50.onnx”模型,用CANN提供的ATC工具将其转换为昇腾AI处理器能识别的OM模型。
1
|
atc --model=resnet50.onnx --framework=5 --output=resnet50 --input_shape="actual_input_1:1,3,224,224" --soc_version=<soc_version> |
atc命令中各参数的含义如下:
- --model:ResNet50的ONNX模型文件存储路径。
- --framework:原始框架类型,5表示ONNX。
- --output:离线OM模型的路径以及文件名,此处文件名配置为resnet50,直接存储在当前目录下。
- --input_shape:执行推理时模型输入数据的shape。
- --soc_version:昇腾AI处理器的型号。
请在服务器种执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的<soc_version>值为Ascendxxxyy。
父主题: 快速入门