summaryrefslogtreecommitdiff
path: root/src/text_recognizer/networks/vqvae/__init__.py
blob: 763953cac4188d2e4618264d0c38624bfd060eca (plain)
1
2
3
4
5
"""VQ-VAE module."""
from .decoder import Decoder
from .encoder import Encoder
from .vector_quantizer import VectorQuantizer
from .vqvae import VQVAE