From 4a54d7e690897dd6e6c719fb908fd371a44c2952 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Sun, 24 Jan 2021 22:14:17 +0100
Subject: Many updates, cool stuff on the way.

---
 src/text_recognizer/networks/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/text_recognizer/networks/__init__.py')

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",
 ]
-- 
cgit v1.2.3-70-g09d2