summaryrefslogtreecommitdiff
path: root/training/conf/network/convnext.yaml
blob: 40343a7943bdc3cce7c267794f39129584c3a2ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
_target_: text_recognizer.network.convnext.convnext.ConvNext
dim: 8
dim_mults: [2, 8]
depths: [2, 2]
attn:
  _target_: text_recognizer.network.convnext.transformer.Transformer
  attn:
    _target_: text_recognizer.network.convnext.transformer.Attention
    dim: 64
    heads: 4
    dim_head: 64
    scale: 8
  ff:
    _target_: text_recognizer.network.convnext.transformer.FeedForward
    dim: 64
    mult: 4