summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae_pixelcnn.yaml
blob: 10200bce5738a58e99a8bd54d6072c5f3bac6489 (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: 512
decay: 0.99