From 91c4a6a90510a9efd492ba41dea3f93b1e58006e Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 25 Jun 2021 12:44:46 +0200
Subject: Add missing zero pad

---
 text_recognizer/networks/encoders/efficientnet/efficientnet.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'text_recognizer')

diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
index b527d90..283b686 100644
--- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py
+++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
@@ -47,6 +47,7 @@ class EfficientNet(nn.Module):
         in_channels = 1  # BW
         out_channels = round_filters(32, self.arch)
         self._conv_stem = nn.Sequential(
+            nn.ZeroPad2d((0, 1, 0, 1)),
             nn.Conv2d(
                 in_channels=in_channels,
                 out_channels=out_channels,
-- 
cgit v1.2.3-70-g09d2