Python sample fix

tomdol · Sep 6, 2021 · eb5890b · eb5890b
1 parent 430ec4f
commit eb5890b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/..._bridges/python/sample/ngraph_function_creation_sample/ngraph_function_creation_sample.py b/..._bridges/python/sample/ngraph_function_creation_sample/ngraph_function_creation_sample.py
@@ -85,7 +85,7 @@ def shape_and_length(shape: list) -> typing.Tuple[list, int]:
     add_1_node = ngraph.add(conv_1_node, add_1_kernel)
 
     # maxpool 1
-    maxpool_1_node = ngraph.max_pool(add_1_node, [2, 2], padding_begin, padding_end, [2, 2], 'ceil', None)
+    maxpool_1_node = ngraph.max_pool(add_1_node, [2, 2], [1, 1], padding_begin, padding_end, [2, 2], 'ceil')
 
     # convolution 2
     conv_2_kernel_shape, conv_2_kernel_length = shape_and_length([50, 20, 5, 5])
@@ -104,7 +104,7 @@ def shape_and_length(shape: list) -> typing.Tuple[list, int]:
     add_2_node = ngraph.add(conv_2_node, add_2_kernel)
 
     # maxpool 2
-    maxpool_2_node = ngraph.max_pool(add_2_node, [2, 2], padding_begin, padding_end, [2, 2], 'ceil', None)
+    maxpool_2_node = ngraph.max_pool(add_2_node, [2, 2], [1, 1], padding_begin, padding_end, [2, 2], 'ceil')
 
     # reshape 1
     reshape_1_dims, reshape_1_length = shape_and_length([2])