Delete convnext

1 files changed, 0 insertions, 21 deletions

diff --git a/text_recognizer/network/convnext/downsample.py b/text_recognizer/network/convnext/downsample.py
deleted file mode 100644
index a8a0466..0000000
--- a/text_recognizer/network/convnext/downsample.py
+++ /dev/null
@@ -1,21 +0,0 @@
-"""Convnext downsample module."""
-from typing import Tuple
-
-from einops.layers.torch import Rearrange
-from torch import Tensor, nn
-
-
-class Downsample(nn.Module):
-    """Downsamples feature maps by patches."""
-
-    def __init__(self, dim: int, dim_out: int, factors: Tuple[int, int]) -> None:
-        super().__init__()
-        s1, s2 = factors
-        self.fn = nn.Sequential(
-            Rearrange("b c (h s1) (w s2) -> b (c s1 s2) h w", s1=s1, s2=s2),
-            nn.Conv2d(dim * s1 * s2, dim_out, 1),
-        )
-
-    def forward(self, x: Tensor) -> Tensor:
-        """Applies patch function."""
-        return self.fn(x)