summaryrefslogtreecommitdiff
path: root/training/conf/network/conv_transformer.yaml
blob: d3a3b0fdbf2707cf7568fd582409a134fb5b12f3 (plain)
1
2
3
4
5
6
7
8
9
10
defaults:
  - encoder: efficientnet
  - decoder: transformer_decoder

_target_: text_recognizer.networks.conv_transformer.ConvTransformer
input_dims: [1, 576, 640]
hidden_dim: 128
dropout_rate: 0.2
num_classes: 1006
pad_index: 1002