summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_transformer.yaml
blob: 5341d8e9f13dbb9cf28e8ae143d0cf4b6c1d7a8e (plain)
1
2
3
4
5
6
7
8
9
10
defaults:
  - mapping: word_piece

_target_: text_recognizer.models.transformer.TransformerLitModel
interval: null
monitor: val/loss
ignore_tokens: [ <s>, <e>, <p> ]
start_token: <s>
end_token: <e>
pad_token: <p>