Skip to content

Commit 686c964

Browse files
committed
fix mindspore conv2d
1 parent 5b816ff commit 686c964

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayerx/backend/ops/mindspore_nn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -537,8 +537,8 @@ def __init__(self, strides, padding, data_format='NHWC', dilations=None, out_cha
537537
self._dilation = (dilations[2], dilations[3])
538538

539539
self.conv2d = P.Conv2D(
540-
out_channel=out_channel, kernel_size=k_size, pad_mode=self.padding, stride=self.ms_stride,
541-
dilation=self.ms_dilation, mode=1, group=1, data_format=self.data_format
540+
out_channel=out_channel, kernel_size=k_size, pad_mode=self.padding, stride=self._stride,
541+
dilation=self._dilation, mode=1, group=1, data_format=self.data_format
542542
)
543543

544544
def construct(self, inputs, filters):

0 commit comments

Comments
 (0)