diff options
Diffstat (limited to 'src/text_recognizer/networks/transformer/__init__.py')
| -rw-r--r-- | src/text_recognizer/networks/transformer/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/text_recognizer/networks/transformer/__init__.py b/src/text_recognizer/networks/transformer/__init__.py index 020a917..9febc88 100644 --- a/src/text_recognizer/networks/transformer/__init__.py +++ b/src/text_recognizer/networks/transformer/__init__.py @@ -1,3 +1,3 @@ """Transformer modules.""" from .positional_encoding import PositionalEncoding -from .transformer import Decoder, Encoder, Transformer +from .transformer import Decoder, Encoder, EncoderLayer, Transformer |