Fixing lint

zhiqwang · Jul 30, 2021 · d941d94 · d941d94
1 parent 083e665
commit d941d94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolort/models/common.py b/yolort/models/common.py
@@ -175,7 +175,7 @@ def space_to_depth(x: Tensor) -> Tensor:
     N, C, H, W = x.size()
     x = x.reshape(N, C, H // 2, 2, W // 2, 2)
     x = x.permute(0, 5, 3, 1, 2, 4)
-    y = x.reshape(N, C*4, H // 2, W // 2)
+    y = x.reshape(N, C * 4, H // 2, W // 2)
     return y