diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-28 21:37:23 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-28 21:37:23 +0100 |
| commit | 5e17ad65db91b26a2f7e66ce2dff8524e4599269 (patch) | |
| tree | cc3bab047aacb13e217eeb90d05d073ee8b3e1a1 | |
| parent | 662f3c6559e23914309f63472035355f4098091c (diff) | |
Update config for lines
| -rw-r--r-- | training/conf/experiment/conv_transformer_lines.yaml | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/training/conf/experiment/conv_transformer_lines.yaml b/training/conf/experiment/conv_transformer_lines.yaml index 38b72c4..1905fcf 100644 --- a/training/conf/experiment/conv_transformer_lines.yaml +++ b/training/conf/experiment/conv_transformer_lines.yaml @@ -109,14 +109,14 @@ network: _target_: text_recognizer.networks.transformer.embeddings.axial.AxialPositionalEmbedding dim: *hidden_dim shape: &shape [3, 64] - axial_encoder: null - # _target_: text_recognizer.networks.transformer.axial_attention.encoder.AxialEncoder - # dim: *hidden_dim - # heads: 4 - # shape: *shape - # depth: 2 - # dim_head: 64 - # dim_index: 1 + axial_encoder: + _target_: text_recognizer.networks.transformer.axial_attention.encoder.AxialEncoder + dim: *hidden_dim + heads: 4 + shape: *shape + depth: 2 + dim_head: 64 + dim_index: 1 model: _target_: text_recognizer.models.transformer.TransformerLitModel |