diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:16:32 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:17:56 +0200 |
| commit | 6e1ad65edd7cbb0f8eb7a48991e9f000f554761d (patch) | |
| tree | fc6ad52d2447928a25d2298e21a9b8ec261b8d1e /text_recognizer/networks/conformer/block.py | |
| parent | 0982e09066a3c31cb8b2fc32b5ecbc2bb64952fb (diff) | |
Rename mlp to ff
Rename mlp to ff
Diffstat (limited to 'text_recognizer/networks/conformer/block.py')
| -rw-r--r-- | text_recognizer/networks/conformer/block.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/networks/conformer/block.py b/text_recognizer/networks/conformer/block.py index d9782e8..4b31aec 100644 --- a/text_recognizer/networks/conformer/block.py +++ b/text_recognizer/networks/conformer/block.py @@ -5,7 +5,7 @@ from typing import Optional from torch import nn, Tensor from text_recognizer.networks.conformer.conv import ConformerConv -from text_recognizer.networks.conformer.mlp import MLP +from text_recognizer.networks.conformer.ff import Feedforward from text_recognizer.networks.conformer.scale import Scale from text_recognizer.networks.transformer.attention import Attention from text_recognizer.networks.transformer.norm import PreNorm @@ -15,7 +15,7 @@ class ConformerBlock(nn.Module): def __init__( self, dim: int, - ff: MLP, + ff: Feedforward, attn: Attention, conv: ConformerConv, ) -> None: |