Fix minor

open-mmlab · Feb 22, 2023 · 7da25d5 · 7da25d5
1 parent 3b915d4
commit 7da25d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmedit/models/common/downsample.py b/mmedit/models/common/downsample.py
@@ -17,6 +17,6 @@ def pixel_unshuffle(x, scale):
             f'with shape: {x.shape}')
     h = h // scale
     w = w // scale
-    x = x.view([b, c, h, scale, w, scale])
+    x = x.view(b, c, h, scale, w, scale)
     x = x.permute(0, 1, 3, 5, 2, 4)
     return x.reshape(b, -1, h, w)