diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-25 12:44:46 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-25 12:44:46 +0200 |
| commit | 91c4a6a90510a9efd492ba41dea3f93b1e58006e (patch) | |
| tree | 2726d5cbe9d855f471ea5b79df5ad0b266163705 /text_recognizer/networks/encoders | |
| parent | e9753c0c1476d4c5aa614e1f65a8dd4302a1ce5b (diff) | |
Add missing zero pad
Diffstat (limited to 'text_recognizer/networks/encoders')
| -rw-r--r-- | text_recognizer/networks/encoders/efficientnet/efficientnet.py | 1 |
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, |