1.软硬件版本

• cann：5.0.2
• atlas200 AI 加速模块

2.操作流程

使用pytorchtoonnx工具，将自己训练出来的模型（.pt文件）转换为onnx文件，再将onnx文件使用ATC工具转换成om文件，提示模型转换成功，但是转换后的模型无法使用可视化工具打开。

使用的框架版本：torch==1.5.0 torchvision==0.6.0 onnx==1.12.0

使用的ATC命令：atc --input_shape="input_features:1,18,3,150" --input_format=NCHW --output="./stgcn_output" --soc_version=Ascend310 --framework=5 --model=./stgcn.onnx

3.报错截图

转换成功 Netron 打开报错

4.问题

1.是否是因为输入--input_shape="input_features:1,18,3,150" 和 --input_format=NCHW 格式没有对应，（N：数量，C:通道，H:高,W:宽）

2.是否是因为ATC还需要其他的命令参数（参数是啥，aipp？）