From a1d795bf02d14befc62cf600fb48842958148eba Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 23 Jul 2021 14:55:31 +0200 Subject: Complete cnn-transformer network, not tested --- text_recognizer/networks/encoders/efficientnet/efficientnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/networks/encoders/efficientnet') diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py index 59598b5..6719efb 100644 --- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py +++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py @@ -41,7 +41,7 @@ class EfficientNet(nn.Module): self.bn_momentum = bn_momentum self.bn_eps = bn_eps self._conv_stem: nn.Sequential = None - self._blocks: nn.Sequential = None + self._blocks: nn.ModuleList = None self._conv_head: nn.Sequential = None self._build() -- cgit v1.2.3-70-g09d2