diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-18 17:42:39 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-18 17:42:39 +0200 |
| commit | 0b8924f37fbab57a3d6f59421e9cd16421c9af4b (patch) | |
| tree | 76f822b08fc16bd373669eab4d8af56269103bc2 /text_recognizer/networks/vqvae/encoder.py | |
| parent | 5fca1bb6aeb48f106750d66b46317acf38cd8a10 (diff) | |
Add more residual blocks in vqvae
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], |