diff options
Diffstat (limited to 'text_recognizer/networks/vqvae/encoder.py')
| -rw-r--r-- | text_recognizer/networks/vqvae/encoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/vqvae/encoder.py b/text_recognizer/networks/vqvae/encoder.py index 06e6c18..fe5ef4b 100644 --- a/text_recognizer/networks/vqvae/encoder.py +++ b/text_recognizer/networks/vqvae/encoder.py @@ -57,7 +57,7 @@ class Encoder(nn.Module): activation_fn, ] - for _ in range(2): + for _ in range(4): encoder += [ Residual( in_channels=self.hidden_dim * self.channels_multipliers[-1], |