diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 00:11:47 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 00:11:47 +0200 |
commit | 376b8d809608618c9a423a1e7849ceee953421f3 (patch) | |
tree | edf89d07915b624354e74c0adf885d43502511c6 /training/conf | |
parent | a4546512c9a8dec632c94e506e1603d460ff0635 (diff) |
Update conv transformer conf
Diffstat (limited to 'training/conf')
-rw-r--r-- | training/conf/network/conv_transformer.yaml | 3 |
1 files changed, 2 insertions, 1 deletions
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 |