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: 512
embedding_dim: 64
num_embeddings: 4096
decay: 0.99