summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae_pixelcnn.yaml
blob: cd850af1f4ad63177008453b484ba90d283c2904 (plain)
1
2
3
4
5
6
7
8
9
defaults:
  - encoder: pixelcnn_encoder
  - decoder: pixelcnn_decoder

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