From 9d49e9514e166a07286196e167b2c486bb9d1ade Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Tue, 28 May 2019 16:42:37 +0800
Subject: [PATCH] fix onnx squeeze

---
 mace/python/tools/converter_tool/onnx_converter.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mace/python/tools/converter_tool/onnx_converter.py b/mace/python/tools/converter_tool/onnx_converter.py
index 70494ae7..ee7a523c 100644
--- a/mace/python/tools/converter_tool/onnx_converter.py
+++ b/mace/python/tools/converter_tool/onnx_converter.py
@@ -1395,8 +1395,8 @@ class OnnxConverter(base_converter.ConverterInterface):
             op.type = MaceOp.Squeeze.name
             axis_arg = op.arg.add()
             axis_arg.name = MaceKeyword.mace_axis_str
-            if 'axis' in node.attrs:
-                axis_value = node.attrs['axis']
+            if 'axes' in node.attrs:
+                axis_value = node.attrs['axes']
             else:
                 axis_value = []
             axis_arg.ints.extend(axis_value)
-- 
GitLab