summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-25 12:47:13 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-25 12:47:13 +0200
commit1aa897cd27990e94ae95e1a9f2ebea035b31d7ea (patch)
treecc2e0b1c4a84b3baa33f912da5db0d313a8625e4
parent0f6f58acd0e548ac8478b93cc6d5e3b0b6ee4a99 (diff)
Fix import of efficientnet module
-rw-r--r--text_recognizer/networks/encoders/__init__.py2
-rw-r--r--text_recognizer/networks/encoders/efficientnet/__init__.py2
2 files changed, 3 insertions, 1 deletions
diff --git a/text_recognizer/networks/encoders/__init__.py b/text_recognizer/networks/encoders/__init__.py
index b526b0c..25aed0e 100644
--- a/text_recognizer/networks/encoders/__init__.py
+++ b/text_recognizer/networks/encoders/__init__.py
@@ -1,2 +1,2 @@
"""Vision backbones."""
-# from .efficientnet import EfficientNet
+from .efficientnet import EfficientNet
diff --git a/text_recognizer/networks/encoders/efficientnet/__init__.py b/text_recognizer/networks/encoders/efficientnet/__init__.py
index e69de29..344233f 100644
--- a/text_recognizer/networks/encoders/efficientnet/__init__.py
+++ b/text_recognizer/networks/encoders/efficientnet/__init__.py
@@ -0,0 +1,2 @@
+"""Efficient net."""
+from .efficientnet import EfficientNet