p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!26447 conv2d pad_mode lower error 

Merge pull request !26447 from chentangyu/code_cty1_master_I4IESU
This commit is contained in:
i-robot 2021-11-18 03:17:14 +00:00 committed by Gitee
parent 707e7fbbf9 a730005dfe
commit 4d1266328b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ops/operations/nn_ops.py
View File

@ -1401,11 +1401,11 @@ class Conv2D(Primitive):
pad = (pad,) * 4
else:
validator.check_equal_int(len(pad), 4, 'pad size', self.name)
self.pad_mode = validator.check_string(pad_mode.lower(), ['valid', 'same', 'pad'], 'pad_mode', self.name)
self.pad_mode = validator.check_string(pad_mode, ['valid', 'same', 'pad'], 'pad_mode', self.name)
if pad_mode != 'pad' and pad != (0, 0, 0, 0):
raise ValueError(f"For '{self.name}', the 'pad' must be zero when 'pad_mode' is not 'pad', "
f"but got 'pad': {self.pad} and 'pad_mode': {pad_mode}.")
f"but got 'pad': {self.pad} and 'pad_mode': {self.pad_mode}.")
self.add_prim_attr("pad", pad)
self.padding = pad
if self.pad_mode == 'pad':