From a30acba378f6f8fb86bb1a6096b9eb0ae7a68449 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 22 Jun 2022 22:27:17 +0200 Subject: Update configs --- training/conf/network/conv_transformer.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'training/conf/network/conv_transformer.yaml') diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index 016416c..24857c8 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 128 +hidden_dim: &hidden_dim 144 num_classes: 58 pad_index: 3 encoder: @@ -46,4 +46,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [18, 80] + shape: [18, 79] -- cgit v1.2.3-70-g09d2