From 89bfe52563582c7e8e4d02c385023b94a25e4bfb Mon Sep 17 00:00:00 2001 From: vinitra Date: Mon, 4 May 2020 16:25:54 -0700 Subject: [PATCH] fix ceiling-mode defaults for pool operators --- .../convert/coreml/operator_converters/neural_network/Pool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxmltools/convert/coreml/operator_converters/neural_network/Pool.py b/onnxmltools/convert/coreml/operator_converters/neural_network/Pool.py index c382dd30..5edc71c9 100644 --- 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