diff options
| -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]: |