diff options
Diffstat (limited to 'src/text_recognizer/networks/residual_network.py')
| -rw-r--r-- | src/text_recognizer/networks/residual_network.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/text_recognizer/networks/residual_network.py b/src/text_recognizer/networks/residual_network.py index e397224..c33f419 100644 --- a/src/text_recognizer/networks/residual_network.py +++ b/src/text_recognizer/networks/residual_network.py @@ -221,8 +221,8 @@ class ResidualNetworkEncoder(nn.Module): nn.Conv2d( in_channels=in_channels, out_channels=self.block_sizes[0], - kernel_size=3, - stride=1, + kernel_size=7, + stride=2, padding=1, bias=False, ), |