mirror of https://github.com/open-mmlab/mmocr.git
17 lines
380 B
Python
17 lines
380 B
Python
|
_base_ = '_base_abinet-vision.py'
|
||
|
|
|
||
|
|
model = dict(
|
||
|
|
decoder=dict(
|
||
|
|
d_model=512,
|
||
|
|
num_iters=3,
|
||
|
|
language_decoder=dict(
|
||
|
|
type='ABILanguageDecoder',
|
||
|
|
d_model=512,
|
||
|
|
n_head=8,
|
||
|
|
d_inner=2048,
|
||
|
|
n_layers=4,
|
||
|
|
dropout=0.1,
|
||
|
|
detach_tokens=True,
|
||
|
|
use_self_attn=False,
|
||
|
|
)), )
|