diff --git a/blueoil/converter/core/operators.py b/blueoil/converter/core/operators.py index 3a9cf5662..480704f9c 100644 --- a/blueoil/converter/core/operators.py +++ b/blueoil/converter/core/operators.py @@ -1041,8 +1041,8 @@ def _check_consistency(self) -> None: mW += f'\nThe weight format is {self._input_ops["W"].dimension}.' self._assert(in_W == self.kernel_width, mW) if self.kernel_index_H is not None and self.index_H is not None: - pad_H = self.pads[self.kernel_index_H] + \ - self.pads[self.kernel_index_H + self._num_dimensions] + pad_H = self.pads[self.kernel_index_H * 2] + \ + self.pads[self.kernel_index_H * 2 + 1] stride_H = self.strides[self.kernel_index_H] dilation_H = self.dilations[self.kernel_index_H] # print(self.name, ' input dimension: ', self.input_ops['X'].dimension) @@ -1068,8 +1068,8 @@ def _check_consistency(self) -> None: print(f'mispadding height at {self.name}: {output_H_rest}') if self.kernel_index_W is not None and self.index_W is not None: - pad_W = self.pads[self.kernel_index_W] + \ - self.pads[self.kernel_index_W + self._num_dimensions] + pad_W = self.pads[self.kernel_index_W * 2] + \ + self.pads[self.kernel_index_W * 2 + 1] stride_W = self.strides[self.kernel_index_W] dilation_W = self.dilations[self.kernel_index_W] # print(self.name, ' input shape: ', self.input_ops['X'].shape) diff --git a/tests/converter/test_consistency_check.py b/tests/converter/test_consistency_check.py index f564a8597..8e10729b2 100644 --- a/tests/converter/test_consistency_check.py +++ b/tests/converter/test_consistency_check.py @@ -114,7 +114,7 @@ def test_conv_consistency(self) -> None: [1, 3, 3, 3], Float32(), input_ops, - pads=[1, 1, 2, 2], + pads=[1, 2, 1, 2], strides=[2, 2] )