mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 11:46:29 +08:00
0aa5d7be6d
* [Model] Add SVTR framework and configs * update * update transform names * update base config * fix cfg * update cfgs * fix * update cfg * update decoder * fix encoder * fix encoder * fix * update cfg * update name
11 lines
218 B
Python
11 lines
218 B
Python
_base_ = [
|
|
'svtr-tiny_20e_st_mj.py',
|
|
]
|
|
|
|
model = dict(
|
|
encoder=dict(
|
|
embed_dims=[96, 192, 256],
|
|
depth=[3, 6, 6],
|
|
num_heads=[3, 6, 8],
|
|
mixer_types=['Local'] * 8 + ['Global'] * 7))
|