summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 18:46:35 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 18:46:49 +0200
commit0bb44f549442ca98d7f57ec825b0c0264a1c069b (patch)
tree97d5e361d9b37315058db298ee2de11ff8c1ea3a
parent764a14468ab8b669dd0906641b891b9bdec7b96a (diff)
Remove import
Remove imports
-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,
-)