summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-07 08:56:52 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-07 08:56:52 +0200
commit2150fd365f9512d55c78cdbef096b7240f619b21 (patch)
tree10557764ac0560b96becd2c00de91004beb8fd04 /text_recognizer/networks/vqvae
parent484dc2b09c87729b4e777e94efdd2e7583651df9 (diff)
Lint
Diffstat (limited to 'text_recognizer/networks/vqvae')
-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: