diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-04 15:15:26 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-04 15:15:26 +0200 |
commit | 04c40f790e405ced6e6b90cf0a8aea268b9345c4 (patch) | |
tree | d5e05ee09fa99ee8d56d5373bde18626274a1fdd /training/conf/network | |
parent | d3afa310f77f47553586eeee58e3d3345a754e2c (diff) |
Add char htr experiment, rename from ocr to htr, vqvae loss collapses
Diffstat (limited to 'training/conf/network')
-rw-r--r-- | training/conf/network/decoder/transformer_decoder.yaml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/network/decoder/transformer_decoder.yaml b/training/conf/network/decoder/transformer_decoder.yaml index c326c04..bc0678b 100644 --- a/training/conf/network/decoder/transformer_decoder.yaml +++ b/training/conf/network/decoder/transformer_decoder.yaml @@ -4,7 +4,7 @@ defaults: _target_: text_recognizer.networks.transformer.Decoder dim: 128 depth: 2 -num_heads: 8 +num_heads: 4 attn_fn: text_recognizer.networks.transformer.attention.Attention attn_kwargs: dim_head: 64 |