summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae.yaml
blob: 936e575bc0e64c220137af57a4ff76f76600f03c (plain)
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: 128
embedding_dim: 64
num_embeddings: 2048
decay: 0.99