diff options
Diffstat (limited to 'text_recognizer/networks/vqvae/decoder.py')
| -rw-r--r-- | text_recognizer/networks/vqvae/decoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/vqvae/decoder.py b/text_recognizer/networks/vqvae/decoder.py index 1194573..5279cbd 100644 --- a/text_recognizer/networks/vqvae/decoder.py +++ b/text_recognizer/networks/vqvae/decoder.py @@ -31,7 +31,7 @@ class Decoder(nn.Module): def _build_decompression_block(self,) -> nn.Sequential: in_channels = self.hidden_dim * self.channels_multipliers[0] decoder = [] - for _ in range(2): + for _ in range(4): decoder += [ Residual( in_channels=in_channels, |