diff options
Diffstat (limited to 'text_recognizer/networks')
| -rw-r--r-- | text_recognizer/networks/vqvae/vqvae.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/text_recognizer/networks/vqvae/vqvae.py b/text_recognizer/networks/vqvae/vqvae.py index 2d17e0f..d876ca1 100644 --- a/text_recognizer/networks/vqvae/vqvae.py +++ b/text_recognizer/networks/vqvae/vqvae.py @@ -29,9 +29,7 @@ class VQVAE(nn.Module): in_channels=embedding_dim, out_channels=hidden_dim, kernel_size=1 ) self.quantizer = VectorQuantizer( - num_embeddings=num_embeddings, - embedding_dim=embedding_dim, - decay=decay, + num_embeddings=num_embeddings, embedding_dim=embedding_dim, decay=decay, ) def encode(self, x: Tensor) -> Tensor: |