Skip to content

Commit 9460481

Browse files
authored
fix ceiling-mode defaults for pool operators (AvgPool, MaxPool) (#388)
1 parent 1e7d1a3 commit 9460481

File tree

1 file changed

+2
-2
lines changed
  • onnxmltools/convert/coreml/operator_converters/neural_network

1 file changed

+2
-2
lines changed

onnxmltools/convert/coreml/operator_converters/neural_network/Pool.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def convert_pooling(scope, operator, container):
186186
op_version = 8
187187
else:
188188
op_version = 10
189-
attrs['ceil_mode'] = 1
189+
attrs['ceil_mode'] = 0
190190
elif params.type == Params.AVERAGE:
191191
op_type = 'AveragePool'
192192
if container.target_opset < 7:
@@ -195,7 +195,7 @@ def convert_pooling(scope, operator, container):
195195
op_version = 7
196196
else:
197197
op_version = 10
198-
attrs['ceil_mode'] = 1
198+
attrs['ceil_mode'] = 0
199199
elif params.type == Params.L2:
200200
op_type = 'LpPool'
201201
attrs['p'] = 2

0 commit comments

Comments
 (0)