Skip to content

Commit 9f1fe6a

Browse files
author
ChongWei905
committed
fix: change ops.pad input format which changed in mindspore in Nov 2022
1 parent 86101cf commit 9f1fe6a

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

mindcv/models/halonet.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -150,10 +150,10 @@ def rel_logits_1d(q, rel_k, permute_mask):
150150
x = msnp.tensordot(q, rel_k, axes=1)
151151
x = ops.reshape(x, (-1, W, rel_size))
152152
# pad to shift from relative to absolute indexing
153-
x_pad = ops.pad(x, paddings=((0, 0), (0, 0), (0, 1)))
153+
x_pad = ops.pad(x, padding=(0, 1))
154154
x_pad = ops.flatten(x_pad)
155155
x_pad = ops.expand_dims(x_pad, 1)
156-
x_pad = ops.pad(x_pad, paddings=((0, 0), (0, 0), (0, rel_size - W)))
156+
x_pad = ops.pad(x_pad, padding=(0, rel_size - W))
157157
x_pad = ops.squeeze(x_pad, axis=())
158158
# reshape adn slice out the padded elements
159159
x_pad = ops.reshape(x_pad, (-1, W+1, rel_size))

mindcv/models/repvgg.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ def get_equivalent_kernel_bias(self):
145145
def _pad_1x1_to_3x3_tensor(self, kernel1x1):
146146
if kernel1x1 is None:
147147
return 0
148-
return ops.pad(kernel1x1, ((1, 1), (1, 1)))
148+
return ops.pad(kernel1x1, (1, 1, 1, 1))
149149

150150
def _fuse_bn_tensor(self, branch):
151151
if branch is None:

mindcv/models/volo.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def construct(self, x: Tensor) -> Tensor:
151151

152152
h = int((H - 1) / self.stride + 1)
153153
w = int((W - 1) / self.stride + 1)
154-
v = ops.pad(v, ((0, 0), (0, 0), (1, 1), (1, 1)))
154+
v = ops.pad(v, (1, 1, 1, 1))
155155
v = self.unfold(v)
156156
v = ops.reshape(v, (B, self.num_heads, C // self.num_heads, self.kernel_size * self.kernel_size, h * w))
157157
v = ops.transpose(v, (0, 1, 4, 3, 2)) # B,H,N,kxk,C/H

0 commit comments

Comments
 (0)