From 87fdd6560854c35962131afb86c3efd05c29aa38 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 22 Nov 2021 22:58:46 +0100 Subject: Fix import efficientnet --- text_recognizer/networks/efficientnet/__init__.py | 3 +-- text_recognizer/networks/efficientnet/mbconv.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) (limited to 'text_recognizer/networks/efficientnet') 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]: -- cgit v1.2.3-70-g09d2