summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/__init__.py3
-rw-r--r--text_recognizer/networks/transformer/__init__.py3
2 files changed, 0 insertions, 6 deletions
diff --git a/text_recognizer/networks/__init__.py b/text_recognizer/networks/__init__.py
index 99f5d1d..f921882 100644
--- a/text_recognizer/networks/__init__.py
+++ b/text_recognizer/networks/__init__.py
@@ -1,5 +1,2 @@
"""Network modules"""
-from text_recognizer.networks.conv_perceiver import ConvPerceiver
from text_recognizer.networks.conv_transformer import ConvTransformer
-from text_recognizer.networks.efficientnet.efficientnet import EfficientNet
-from text_recognizer.networks.vq_transformer import VqTransformer
diff --git a/text_recognizer/networks/transformer/__init__.py b/text_recognizer/networks/transformer/__init__.py
index 73c18d1..af08aa8 100644
--- a/text_recognizer/networks/transformer/__init__.py
+++ b/text_recognizer/networks/transformer/__init__.py
@@ -4,6 +4,3 @@ from text_recognizer.networks.transformer.ff import FeedForward
from text_recognizer.networks.transformer.norm import RMSNorm
from text_recognizer.networks.transformer.decoder import Decoder, DecoderBlock
from text_recognizer.networks.transformer.embeddings.rotary import RotaryEmbedding
-from text_recognizer.networks.transformer.embeddings.axial import (
- AxialPositionalEmbedding,
-)