diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-07 08:56:52 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-07 08:56:52 +0200 |
| commit | 2150fd365f9512d55c78cdbef096b7240f619b21 (patch) | |
| tree | 10557764ac0560b96becd2c00de91004beb8fd04 /text_recognizer/networks/vqvae | |
| parent | 484dc2b09c87729b4e777e94efdd2e7583651df9 (diff) | |
Lint
Diffstat (limited to 'text_recognizer/networks/vqvae')
| -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: |