From a347b7e08df174d10d2cdffc269f42a2a03490a2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 19 Jun 2022 19:55:59 +0200 Subject: Update configs --- training/conf/network/conv_transformer.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (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 39c5c46..ccdf960 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -11,6 +11,7 @@ encoder: bn_eps: 1.0e-3 depth: 3 out_channels: *hidden_dim + stride: [2, 1] decoder: _target_: text_recognizer.networks.transformer.Decoder depth: 6 @@ -46,4 +47,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [72, 80] + shape: [17, 78] -- cgit v1.2.3-70-g09d2