summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_transformer.yaml
blob: 6ffde4ee8cd18530516297a79f56858b443505f7 (plain)
1
2
3
4
5
6
7
_target_: text_recognizer.models.transformer.TransformerLitModel
interval: step
monitor: val/loss
ignore_tokens: [ <s>, <e>, <p> ]
start_token: <s>
end_token: <e>
pad_token: <p>