mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
6ca7404925
* [Config] Add textrec_default_runtime * [Config] Add textrec_default_runtime * add vis hook * update satrn cfg * update * update Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
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))
|