summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae.yaml
blob: 5a5c066f485b8817020f75e06be5868e91eeda2b (plain)
1
2
3
4
5
6
7
8
_target_: text_recognizer.networks.vqvae.VQVAE
in_channels: 1
res_channels: 32
num_residual_layers: 2
embedding_dim: 64
num_embeddings: 512
decay: 0.99
activation: mish