diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-23 14:55:31 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-23 14:55:31 +0200 |
| commit | a1d795bf02d14befc62cf600fb48842958148eba (patch) | |
| tree | 21465c20262b15654985368731e8a289562e8df7 /text_recognizer/networks/encoders/efficientnet/efficientnet.py | |
| parent | d20802e1f412045f7afa4bd8ac50be3488945e90 (diff) | |
Complete cnn-transformer network, not tested
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/efficientnet.py')
| -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() |