summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/efficientnet
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-22 22:58:46 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-22 22:58:46 +0100
commit87fdd6560854c35962131afb86c3efd05c29aa38 (patch)
tree8750b50d716e55b392543755ab00531cbf961fc3 /text_recognizer/networks/efficientnet
parenta3942173791eb82fc3b6a371ad2dbbeb12c1c2b0 (diff)
Fix import efficientnet
Diffstat (limited to 'text_recognizer/networks/efficientnet')
-rw-r--r--text_recognizer/networks/efficientnet/__init__.py3
-rw-r--r--text_recognizer/networks/efficientnet/mbconv.py2
2 files changed, 2 insertions, 3 deletions
diff --git a/text_recognizer/networks/efficientnet/__init__.py b/text_recognizer/networks/efficientnet/__init__.py
index 344233f..66ed961 100644
--- a/text_recognizer/networks/efficientnet/__init__.py
+++ b/text_recognizer/networks/efficientnet/__init__.py
@@ -1,2 +1 @@
-"""Efficient net."""
-from .efficientnet import EfficientNet
+"""Efficientnet module."""
diff --git a/text_recognizer/networks/efficientnet/mbconv.py b/text_recognizer/networks/efficientnet/mbconv.py
index 4b051eb..beb7d57 100644
--- a/text_recognizer/networks/efficientnet/mbconv.py
+++ b/text_recognizer/networks/efficientnet/mbconv.py
@@ -6,7 +6,7 @@ import torch
from torch import nn, Tensor
import torch.nn.functional as F
-from text_recognizer.networks.encoders.efficientnet.utils import stochastic_depth
+from text_recognizer.networks.efficientnet.utils import stochastic_depth
def _convert_stride(stride: Union[Tuple[int, int], int]) -> Tuple[int, int]: