rebase

mindee · Mar 25, 2024 · 7bc3885 · 7bc3885
1 parent a177bf3
commit 7bc3885
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/doctr/models/detection/fast/pytorch.py b/doctr/models/detection/fast/pytorch.py
@@ -58,9 +58,9 @@ def __init__(
         out_channels: int = 128,
     ) -> None:
         super().__init__()
-        self.reduction = nn.ModuleList(
-            [FASTConvLayer(in_channels * scale, out_channels, kernel_size=3) for scale in [1, 2, 4, 8]]
-        )
+        self.reduction = nn.ModuleList([
+            FASTConvLayer(in_channels * scale, out_channels, kernel_size=3) for scale in [1, 2, 4, 8]
+        ])
 
     def _upsample(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
         return F.interpolate(x, size=y.shape[-2:], mode="bilinear")