summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/conformer/mlp.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks/conformer/mlp.py')
-rw-r--r--text_recognizer/networks/conformer/mlp.py17
1 files changed, 0 insertions, 17 deletions
diff --git a/text_recognizer/networks/conformer/mlp.py b/text_recognizer/networks/conformer/mlp.py
deleted file mode 100644
index 031bde9..0000000
--- a/text_recognizer/networks/conformer/mlp.py
+++ /dev/null
@@ -1,17 +0,0 @@
-"""Conformer feedforward block."""
-from torch import nn, Tensor
-
-
-class MLP(nn.Module):
- def __init__(self, dim: int, mult: int = 4, dropout: float = 0.0) -> None:
- super().__init__()
- self.layers = nn.Sequential(
- nn.Linear(dim, mult * dim),
- nn.Mish(inplace=True),
- nn.Dropout(dropout),
- nn.Linear(mult * dim, dim),
- nn.Dropout(dropout),
- )
-
- def forward(self, x: Tensor) -> Tensor:
- return self.layers(x)