mmocr/configs/textrecog/svtr/svtr-small_20e_st_mj.py

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))