Many updates, cool stuff on the way.

1 files changed, 8 insertions, 0 deletions

diff --git a/src/text_recognizer/networks/__init__.py b/src/text_recognizer/networks/__init__.py
index 2b624bb..bac5d28 100644
--- a/src/text_recognizer/networks/__init__.py
+++ b/src/text_recognizer/networks/__init__.py
@@ -1,4 +1,5 @@
 """Network modules."""
+from .cnn import CNN
 from .cnn_transformer import CNNTransformer
 from .crnn import ConvolutionalRecurrentNetwork
 from .ctc import greedy_decoder
@@ -7,15 +8,19 @@ from .lenet import LeNet
 from .metrics import accuracy, cer, wer
 from .mlp import MLP
 from .residual_network import ResidualNetwork, ResidualNetworkEncoder
+from .transducer import TDS2d
 from .transformer import Transformer
 from .unet import UNet
 from .util import sliding_window
 from .vit import ViT
+from .vq_transformer import VQTransformer
+from .vqvae import VQVAE
 from .wide_resnet import WideResidualNetwork
 
 __all__ = [
     "accuracy",
     "cer",
+    "CNN",
     "CNNTransformer",
     "ConvolutionalRecurrentNetwork",
     "DenseNet",
@@ -27,8 +32,11 @@ __all__ = [
     "ResidualNetworkEncoder",
     "sliding_window",
     "UNet",
+    "TDS2d",
     "Transformer",
     "ViT",
+    "VQTransformer",
+    "VQVAE",
     "wer",
     "WideResidualNetwork",
 ]