summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/encoders/efficientnet
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-25 12:44:46 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-25 12:44:46 +0200
commit91c4a6a90510a9efd492ba41dea3f93b1e58006e (patch)
tree2726d5cbe9d855f471ea5b79df5ad0b266163705 /text_recognizer/networks/encoders/efficientnet
parente9753c0c1476d4c5aa614e1f65a8dd4302a1ce5b (diff)
Add missing zero pad
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet')
-rw-r--r--text_recognizer/networks/encoders/efficientnet/efficientnet.py1
1 files changed, 1 insertions, 0 deletions
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,