diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-22 22:58:46 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-22 22:58:46 +0100 |
| commit | 87fdd6560854c35962131afb86c3efd05c29aa38 (patch) | |
| tree | 8750b50d716e55b392543755ab00531cbf961fc3 | |
| parent | a3942173791eb82fc3b6a371ad2dbbeb12c1c2b0 (diff) | |
Fix import efficientnet
| -rw-r--r-- | text_recognizer/networks/efficientnet/__init__.py | 3 | ||||
| -rw-r--r-- | text_recognizer/networks/efficientnet/mbconv.py | 2 |
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]: |