summaryrefslogtreecommitdiff
path: root/training/conf/network/conv_transformer.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-05 00:09:19 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-05 00:09:19 +0200
commit12abf17cd7c31ae4599be366505a4423fbba4044 (patch)
tree996e5d549ebbb7d22f5acfdcd321bddec77f98d1 /training/conf/network/conv_transformer.yaml
parent16e2e420e077253c3b2bc414283281fea557717d (diff)
Update perceiver conf
Diffstat (limited to 'training/conf/network/conv_transformer.yaml')
-rw-r--r--training/conf/network/conv_transformer.yaml2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml
index c4cf13e..f618ba1 100644
--- a/training/conf/network/conv_transformer.yaml
+++ b/training/conf/network/conv_transformer.yaml
@@ -1,6 +1,6 @@
_target_: text_recognizer.networks.ConvTransformer
input_dims: [1, 1, 576, 640]
-hidden_dim: &hidden_dim 96
+hidden_dim: &hidden_dim 144
num_classes: 58
pad_index: 3
encoder: