summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/__init__.py
blob: 4ff48f7bd1130513a7f3f67c95c80f3fd33a3843 (plain)
1
2
3
4
5
6
7
8
"""Transformer modules."""
from .positional_encoding import (
    PositionalEncoding,
    PositionalEncoding2D,
    target_padding_mask,
)

# from .transformer import Decoder, DecoderLayer, Encoder, EncoderLayer, Transformer