training/conf/network/vqvae.yaml


1
2
3
4
5
6
7
8
9

defaults:
  - encoder: vae_encoder
  - decoder: vae_decoder

_target_: text_recognizer.networks.vqvae.vqvae.VQVAE
hidden_dim: 256
embedding_dim: 32
num_embeddings: 1024
decay: 0.99