summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/__init__.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-27 18:17:42 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-27 18:17:42 +0200
commit889731f491266a0626861f07fe2a0de1f6caa468 (patch)
treef627b5c2102036f91fb0c6d20ab622c36412d059 /text_recognizer/networks/__init__.py
parent68cda1e6abf76312eb13f45ca3c8565a0b00d745 (diff)
Add vq transformer network
Diffstat (limited to 'text_recognizer/networks/__init__.py')
-rw-r--r--text_recognizer/networks/__init__.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/networks/__init__.py b/text_recognizer/networks/__init__.py
index dad9d23..e2d6fd5 100644
--- a/text_recognizer/networks/__init__.py
+++ b/text_recognizer/networks/__init__.py
@@ -1,3 +1,4 @@
"""Network modules"""
from text_recognizer.networks.conv_transformer import ConvTransformer
from text_recognizer.networks.efficientnet.efficientnet import EfficientNet
+from text_recognizer.networks.vq_transformer import VqTransformer