diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-08 19:59:55 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-08 19:59:55 +0200 |
| commit | 240f5e9f20032e82515fa66ce784619527d1041e (patch) | |
| tree | b002d28bbfc9abe9b6af090f7db60bea0aeed6e8 /text_recognizer/networks/vqvae/vqvae.py | |
| parent | d12f70402371dda586d457af2a3df7fb5b3130ad (diff) | |
Add VQGAN and loss function
Diffstat (limited to 'text_recognizer/networks/vqvae/vqvae.py')
| -rw-r--r-- | text_recognizer/networks/vqvae/vqvae.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/text_recognizer/networks/vqvae/vqvae.py b/text_recognizer/networks/vqvae/vqvae.py index 0646119..e8660c4 100644 --- a/text_recognizer/networks/vqvae/vqvae.py +++ b/text_recognizer/networks/vqvae/vqvae.py @@ -32,7 +32,6 @@ class VQVAE(nn.Module): num_embeddings=num_embeddings, embedding_dim=embedding_dim, decay=decay, ) - def encode(self, x: Tensor) -> Tensor: """Encodes input to a latent code.""" z_e = self.encoder(x) |