diff options
| -rw-r--r-- | text_recognizer/networks/transformer/decoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/transformer/decoder.py b/text_recognizer/networks/transformer/decoder.py index db5c766..8ea021b 100644 --- a/text_recognizer/networks/transformer/decoder.py +++ b/text_recognizer/networks/transformer/decoder.py @@ -5,7 +5,7 @@ from typing import Optional, Type from torch import nn, Tensor from text_recognizer.networks.transformer.attention import Attention -from text_recognizer.networks.transformer.mlp import FeedForward +from text_recognizer.networks.transformer.ff import FeedForward class DecoderBlock(nn.Module): |