summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/vqvae/vqvae.py4
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: