diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-05 00:06:37 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-05 00:06:37 +0200 |
| commit | cfb460666953c87f606833bf597b53eba0a2900d (patch) | |
| tree | ece2fcb5c9003a4dac8ed4f2108d063b07cdc274 /text_recognizer/networks/efficientnet/efficientnet.py | |
| parent | f95d51e45ea24a956ce4384e4680f849651b2506 (diff) | |
Format
Diffstat (limited to 'text_recognizer/networks/efficientnet/efficientnet.py')
| -rw-r--r-- | text_recognizer/networks/efficientnet/efficientnet.py | 11 |
1 files changed, 0 insertions, 11 deletions
diff --git a/text_recognizer/networks/efficientnet/efficientnet.py b/text_recognizer/networks/efficientnet/efficientnet.py index bd47e4b..3481090 100644 --- a/text_recognizer/networks/efficientnet/efficientnet.py +++ b/text_recognizer/networks/efficientnet/efficientnet.py @@ -73,17 +73,6 @@ class EfficientNet(nn.Module): num_features=out_channels, momentum=self.bn_momentum, eps=self.bn_eps ), nn.Mish(inplace=True), - nn.Conv2d( - in_channels=out_channels, - out_channels=out_channels, - kernel_size=3, - stride=2, - bias=False, - ), - nn.BatchNorm2d( - num_features=out_channels, momentum=self.bn_momentum, eps=self.bn_eps - ), - nn.Mish(inplace=True), ) self._blocks = nn.ModuleList([]) for args in _block_args: |