From d941d9455380e1208766bc527ce714a7b9e6714e Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Fri, 30 Jul 2021 12:57:12 -0400 Subject: [PATCH] Fixing lint --- yolort/models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yolort/models/common.py b/yolort/models/common.py index e3786cf7..4e37705a 100644 --- 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