summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks/vqvae/__init__.py')
-rw-r--r--text_recognizer/networks/vqvae/__init__.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/text_recognizer/networks/vqvae/__init__.py b/text_recognizer/networks/vqvae/__init__.py
new file mode 100644
index 0000000..763953c
--- /dev/null
+++ b/text_recognizer/networks/vqvae/__init__.py
@@ -0,0 +1,5 @@
+"""VQ-VAE module."""
+from .decoder import Decoder
+from .encoder import Encoder
+from .vector_quantizer import VectorQuantizer
+from .vqvae import VQVAE