mirror of https://github.com/open-mmlab/mmocr.git
24 lines
547 B
Python
24 lines
547 B
Python
|
_base_ = ['satrn_shallow_5e_st_mj.py']
|
||
|
|
||
|
|
model = dict(
|
||
|
|
backbone=dict(type='ShallowCNN', input_channels=3, hidden_dim=256),
|
||
|
|
encoder=dict(
|
||
|
type='SATRNEncoder',
|
||
|
|
n_layers=6,
|
||
|
|
n_head=8,
|
||
|
|
d_k=256 // 8,
|
||
|
|
d_v=256 // 8,
|
||
|
|
d_model=256,
|
||
|
|
n_position=100,
|
||
|
|
d_inner=256 * 4,
|
||
|
|
dropout=0.1),
|
||
|
|
decoder=dict(
|
||
|
type='NRTRDecoder',
|
||
|
|
n_layers=6,
|
||
|
|
d_embedding=256,
|
||
|
|
n_head=8,
|
||
|
|
d_model=256,
|
||
|
|
d_inner=256 * 4,
|
||
|
|
d_k=256 // 8,
|
||
|
|
d_v=256 // 8))
|