diff options
Diffstat (limited to 'training')
| -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 |