diff options
Diffstat (limited to 'text_recognizer/networks/encoders')
| -rw-r--r-- | text_recognizer/networks/encoders/efficientnet/efficientnet.py | 2 |
1 files changed, 1 insertions, 1 deletions
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() |