diff --git a/doctr/models/detection/fast/pytorch.py b/doctr/models/detection/fast/pytorch.py index d6da944de2..3ac8af6f49 100644 --- 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")