mmocr-demo / configs /_base_ /recog_models /nrtr_modality_transform.py
Xianbao QIAN
add Dockerfile
378b1f2
raw
history blame
No virus
358 Bytes
label_convertor = dict(
type='AttnConvertor', dict_type='DICT36', with_unknown=True, lower=True)
model = dict(
type='NRTR',
backbone=dict(type='NRTRModalityTransform'),
encoder=dict(type='NRTREncoder', n_layers=12),
decoder=dict(type='NRTRDecoder'),
loss=dict(type='TFLoss'),
label_convertor=label_convertor,
max_seq_len=40)