diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:48:18 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:48:18 +0200 |
commit | e643e0c61ab33ce1bb8cfdebc92fc0670c82afda (patch) | |
tree | 10876c66ed249429e42f1660231bb7ff84c10769 /training/conf/network/convnext.yaml | |
parent | 9ce21f569ecac03f15f2ad669fde3dd4a512f8cc (diff) |
Update configs
Diffstat (limited to 'training/conf/network/convnext.yaml')
-rw-r--r-- | training/conf/network/convnext.yaml | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/training/conf/network/convnext.yaml b/training/conf/network/convnext.yaml index 40343a7..bcbc78e 100644 --- a/training/conf/network/convnext.yaml +++ b/training/conf/network/convnext.yaml @@ -1,15 +1,15 @@ _target_: text_recognizer.network.convnext.convnext.ConvNext dim: 8 -dim_mults: [2, 8] -depths: [2, 2] +dim_mults: [2, 8, 8, 8] +depths: [2, 2, 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 + 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 |