问题现象描述

Mindspore训练的flyspeech模型导出ONNX文件后，使用ATC工具转OM模型时报[Node:176Squeeze] Check input shape faild错误信息。

原因分析

算子输入shape不匹配。

问题排查：

1. 使用以下命令设置日志级别并设置环境变量将ATC日志打印到屏幕。

   --log=debug
   export ASCEND_SLOG_PRINT_TO_STDOUT=1
   export ASCEND_GLOBAL_LOG_LEVEL=0

2. 执行ATC命令并将日志重定向到debug.log。
3. 使用以下命令搜索日志，搜索结果如下图所示。

   grep 176Squeeze debug.log | grep "before_infer"

   

4. 使用Netron软件打开ONNX文件，查看176Squeeze算子信息，如下图所示。

   

   Squeeze算子对输入进行维度压缩，指定压缩的维度大小应为1；上述信息表示176Squeeze需要对输入沿dim[2]进行压缩，但是输入shape为[1,512,8,193]。

   dim[2]!=1导致转换失败。

解决措施

当前支持导出ONNX的模型清单中不包含该模型，所以无法保证当前ONNX文件的正确性，因此建议导出air格式的文件，重新执行ATC命令。