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