vae_encoder.yaml « encoder « network « conf « training - text-recognizer.git - A neural network that reads handwritten paragraphs.

summary refs log tree commit diff

path: root/training/conf/network/encoder/vae_encoder.yaml

blob: 2ea3adf737b3afdc1d4f2612010ebe5bca6ee2d4 (plain)

_target_: text_recognizer.networks.vqvae.encoder.Encoder
in_channels: 1 
hidden_dim: 32
channels_multipliers: [1, 2, 4]
dropout_rate: 0.0
activation: leaky_relu