From 376b8d809608618c9a423a1e7849ceee953421f3 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 27 Sep 2022 00:11:47 +0200 Subject: Update conv transformer conf --- training/conf/network/conv_transformer.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'training/conf') diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index c71296b..0ef862f 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -11,9 +11,10 @@ encoder: downsampling_factors: [[2, 2], [2, 2], [2, 2]] decoder: _target_: text_recognizer.networks.transformer.Decoder + dim: *hidden_dim depth: 10 block: - _target_: text_recognizer.networks.transformer.DecoderBlock + _target_: text_recognizer.networks.transformer.decoder_block.DecoderBlock self_attn: _target_: text_recognizer.networks.transformer.Attention dim: *hidden_dim -- cgit v1.2.3-70-g09d2