summaryrefslogtreecommitdiff
path: root/training/conf/network/convnext.yaml
blob: bcbc78effeb8b4a4541a098c3c5c7e2f3e31841f (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, 8, 8]
depths: [2, 2, 2, 2]
attn:
  _target_: text_recognizer.network.convnext.transformer.Transformer
  attn: null
    # _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