halonet pr

mindspore-lab · Aug 7, 2023 · 94144bd · 94144bd
1 parent 11fc7bc
commit 94144bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindcv/models/halonet.py b/mindcv/models/halonet.py
@@ -150,7 +150,7 @@ def rel_logits_1d(q, rel_k, permute_mask):
     x = msnp.tensordot(q, rel_k, axes=1)
     x = ops.reshape(x, (-1, W, rel_size))
     # pad to shift from relative to absolute indexing
-    x_pad = ops.pad(x,paddings=((0, 0), (0, 0), (0, 1)))
+    x_pad = ops.pad(x, paddings=((0, 0), (0, 0), (0, 1)))
     x_pad = ops.flatten(x_pad)
     x_pad = ops.expand_dims(x_pad, 1)
     x_pad = ops.pad(x_pad, paddings=((0, 0), (0, 0), (0, rel_size - W)))