summaryrefslogtreecommitdiff
path: root/training/conf/network
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 21:58:38 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 21:58:38 +0200
commit7bdb8005f8693938a43d6732300f1c1b17fcbc79 (patch)
tree9ed4a62e224cd849fa80c002c6de04f63df1cce6 /training/conf/network
parent4a4d5f2a2ee06069140b0d861018a70c63ad3d46 (diff)
Update configs
Diffstat (limited to 'training/conf/network')
-rw-r--r--training/conf/network/convnext.yaml12
1 files changed, 12 insertions, 0 deletions
diff --git a/training/conf/network/convnext.yaml b/training/conf/network/convnext.yaml
index bc1ce93..63ad424 100644
--- a/training/conf/network/convnext.yaml
+++ b/training/conf/network/convnext.yaml
@@ -3,3 +3,15 @@ dim: 16
dim_mults: [2, 4, 8]
depths: [3, 3, 6]
downsampling_factors: [[2, 2], [2, 2], [2, 2]]
+attn:
+ _target_: text_recognizer.networks.convnext.TransformerBlock
+ attn:
+ _target_: text_recognizer.networks.convnext.Attention
+ dim: 128
+ heads: 4
+ dim_head: 64
+ scale: 8
+ ff:
+ _target_: text_recognizer.networks.convnext.FeedForward
+ dim: 128
+ mult: 4