text_recognizer/networks/transformer/__init__.py


1
2
3
4
5
6
7

"""Transformer modules."""
from .positional_encoding import (
    PositionalEncoding,
    PositionalEncoding2D,
    target_padding_mask,
)
from .transformer import Decoder, DecoderLayer, Encoder, EncoderLayer, Transformer