Update simmim_neck.py

Lint errors
2024-02-26 15:48:38 +01:00 · 2024-02-26 15:48:38 +01:00 · ecf83c4007
parent c088a191ec
commit ecf83c4007
1 changed files with 2 additions and 1 deletions
--- a/mmpretrain/models/necks/simmim_neck.py
+++ b/mmpretrain/models/necks/simmim_neck.py
@ -18,7 +18,8 @@ class SimMIMLinearDecoder(BaseModule):
        target_channels (int): Channel dimensions of original image.
    """

-    def __init__(self, in_channels: int, encoder_stride: int, target_channels: int = 3) -> None:
+    def __init__(self, in_channels: int, encoder_stride: int,
+                 target_channels: int = 3) -> None:
        super().__init__()
        self.decoder = nn.Sequential(
            nn.Conv2d(