From f9ede1e61008ead9b7abe910dff79067cf862312 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 18 Sep 2021 17:44:19 +0200
Subject: Add comment for quantization
---
text_recognizer/networks/vqvae/vqvae.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/text_recognizer/networks/vqvae/vqvae.py b/text_recognizer/networks/vqvae/vqvae.py
index e8660c4..f31b062 100644
--- a/text_recognizer/networks/vqvae/vqvae.py
+++ b/text_recognizer/networks/vqvae/vqvae.py
@@ -38,6 +38,7 @@ class VQVAE(nn.Module):
return self.pre_codebook_conv(z_e)
def quantize(self, z_e: Tensor) -> Tuple[Tensor, Tensor]:
+ """Quantizes the encoded latent vectors."""
z_q, vq_loss = self.quantizer(z_e)
return z_q, vq_loss
--
cgit v1.2.3-70-g09d2