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: 58e905dfdc7ce491285e78cc876386ff15ea50df (plain)

_target_: text_recognizer.networks.vqvae.encoder.Encoder
in_channels: 1 
hidden_dim: 32
channels_multipliers: [1, 2, 4, 8, 8]
dropout_rate: 0.25