From eb5890b36a2bda8acdd0b9c4d6e412461535991e Mon Sep 17 00:00:00 2001 From: tomdol Date: Mon, 6 Sep 2021 18:40:41 +0200 Subject: [PATCH] Python sample fix --- .../ngraph_function_creation_sample.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/inference-engine/ie_bridges/python/sample/ngraph_function_creation_sample/ngraph_function_creation_sample.py b/inference-engine/ie_bridges/python/sample/ngraph_function_creation_sample/ngraph_function_creation_sample.py index 40499546891056..84e0ed84ed552e 100755 --- a/inference-engine/ie_bridges/python/sample/ngraph_function_creation_sample/ngraph_function_creation_sample.py +++ b/inference-engine/ie_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])