fix ceiling-mode defaults for pool operators (AvgPool, MaxPool) (onnx…

…#388)
jiafatom · May 4, 2020 · 9460481 · 9460481
1 parent 1e7d1a3
commit 9460481
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxmltools/convert/coreml/operator_converters/neural_network/Pool.py b/onnxmltools/convert/coreml/operator_converters/neural_network/Pool.py
@@ -186,7 +186,7 @@ def convert_pooling(scope, operator, container):
             op_version = 8
         else:
             op_version = 10
-            attrs['ceil_mode'] = 1
+            attrs['ceil_mode'] = 0
     elif params.type == Params.AVERAGE:
         op_type = 'AveragePool'
         if container.target_opset < 7:
@@ -195,7 +195,7 @@ def convert_pooling(scope, operator, container):
             op_version = 7
         else:
             op_version = 10
-            attrs['ceil_mode'] = 1
+            attrs['ceil_mode'] = 0
     elif params.type == Params.L2:
         op_type = 'LpPool'
         attrs['p'] = 2