diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:10:23 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:10:23 +0200 |
| commit | 3fa2733b697bfe1c1f8a6031ae7f1f66343d5a5b (patch) | |
| tree | 573269462c0d56860baa287c01c58fb96f892476 /text_recognizer/networks/transformer | |
| parent | 8c69c3d457ed3279c59a8c25b08a49f39562db51 (diff) | |
Update ff import
Diffstat (limited to 'text_recognizer/networks/transformer')
| -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): |