From 4c37420890e3a1efca34838a8c8fb15fca738ffa Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Fri, 24 Apr 2020 15:13:21 +0800 Subject: [PATCH] make conv2d bp filter stride attr len 4 --- mindspore/ops/operations/_grad_ops.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/ops/operations/_grad_ops.py b/mindspore/ops/operations/_grad_ops.py index 782784ca00..c821063da8 100644 --- a/mindspore/ops/operations/_grad_ops.py +++ b/mindspore/ops/operations/_grad_ops.py @@ -174,9 +174,9 @@ class Conv2DBackpropFilter(PrimitiveWithInfer): pad_mode = pad_mode.upper() self.add_prim_attr('pad_mode', pad_mode) self.pad = pad - if isinstance(stride, tuple) and len(stride) == 4: - self.stride = (stride[2], stride[3]) - self.add_prim_attr('stride', self.stride) + if isinstance(stride, tuple) and len(stride) == 2: + self.stride = stride + self.add_prim_attr('stride', (1, 1, self.stride[0], self.stride[1])) self.dilation = dilation self.group = group self.add_prim_attr('data_format', "NCHW")