diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:56:50 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:56:50 +0200 |
commit | 598fd271942884f852e98c82d6caedad425813a1 (patch) | |
tree | 19adb7c3960ecb4079cb02a8ee38812caee7a222 /text_recognizer | |
parent | 892137c4f1104edc9645c0e1e877b885816fa83c (diff) |
Remove strides
Diffstat (limited to 'text_recognizer')
-rw-r--r-- | text_recognizer/networks/efficientnet/efficientnet.py | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/text_recognizer/networks/efficientnet/efficientnet.py b/text_recognizer/networks/efficientnet/efficientnet.py index 2a712d8..c94c748 100644 --- a/text_recognizer/networks/efficientnet/efficientnet.py +++ b/text_recognizer/networks/efficientnet/efficientnet.py @@ -36,7 +36,6 @@ class EfficientNet(nn.Module): bn_eps: float = 1.0e-3, depth: int = 7, out_channels: int = 1280, - stride: Tuple[int, int] = (2, 2), ) -> None: super().__init__() self.params = self._get_arch_params(arch) @@ -44,7 +43,6 @@ class EfficientNet(nn.Module): self.bn_momentum = bn_momentum self.bn_eps = bn_eps self.depth = depth - self.stride = stride self.out_channels: int = out_channels self._conv_stem: nn.Sequential self._blocks: nn.ModuleList |