summaryrefslogtreecommitdiff
path: root/training/conf/network/encoder/vae_encoder.yaml
blob: 5d39bf71718ad8a2b6f05a8784fc1831ce202771 (plain)
1
2
3
4
5
6
7
8
9
_target_: text_recognizer.networks.vqvae.encoder.Encoder
in_channels: 1 
hidden_dim: 32
channels_multipliers: [1, 2, 4]
dropout_rate: 0.0
activation: mish
use_norm: true
num_residuals: 4
residual_channels: 32