From 521f3bbbba9f04f48e81d78033c6e1c29a08e515 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 20 Jun 2022 00:09:20 +0200
Subject: Update eff net config

---
 text_recognizer/networks/efficientnet/efficientnet.py | 17 ++---------------
 1 file changed, 2 insertions(+), 15 deletions(-)

(limited to 'text_recognizer/networks/efficientnet/efficientnet.py')

diff --git a/text_recognizer/networks/efficientnet/efficientnet.py b/text_recognizer/networks/efficientnet/efficientnet.py
index de08457..2a712d8 100644
--- a/text_recognizer/networks/efficientnet/efficientnet.py
+++ b/text_recognizer/networks/efficientnet/efficientnet.py
@@ -61,14 +61,14 @@ class EfficientNet(nn.Module):
         """Builds the efficientnet backbone."""
         _block_args = block_args()[: self.depth]
         in_channels = 1  # BW
-        out_channels = round_filters(32, self.params)
+        out_channels = round_filters(16, self.params)
         self._conv_stem = nn.Sequential(
             nn.ZeroPad2d((0, 1, 0, 1)),
             nn.Conv2d(
                 in_channels=in_channels,
                 out_channels=out_channels,
                 kernel_size=3,
-                stride=self.stride,
+                stride=2,
                 bias=False,
             ),
             nn.BatchNorm2d(
@@ -98,19 +98,6 @@ class EfficientNet(nn.Module):
             nn.Conv2d(
                 in_channels,
                 self.out_channels,
-                kernel_size=2,
-                stride=self.stride,
-                bias=False,
-            ),
-            nn.BatchNorm2d(
-                num_features=self.out_channels,
-                momentum=self.bn_momentum,
-                eps=self.bn_eps,
-            ),
-            nn.Mish(inplace=True),
-            nn.Conv2d(
-                self.out_channels,
-                self.out_channels,
                 kernel_size=1,
                 stride=1,
                 bias=False,
-- 
cgit v1.2.3-70-g09d2