diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-09 22:36:04 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-09 22:36:04 +0200 |
commit | 5c28ad1905f63838afbc7ee6c5ec31ff1310e3a1 (patch) | |
tree | 594796cc7d8bd01fedb167208dbf2593f2b898a1 /training/conf/network/conformer | |
parent | 4c7f720fe42278ec27e87bb1c3d5fcc6f1f822c2 (diff) |
Move conformer config
Diffstat (limited to 'training/conf/network/conformer')
-rw-r--r-- | training/conf/network/conformer/conformer.yaml | 27 |
1 files changed, 0 insertions, 27 deletions
diff --git a/training/conf/network/conformer/conformer.yaml b/training/conf/network/conformer/conformer.yaml deleted file mode 100644 index 2bb423f..0000000 --- a/training/conf/network/conformer/conformer.yaml +++ /dev/null @@ -1,27 +0,0 @@ ---- -_target_: text_recognizer.networks.conformer.Conformer -depth: 16 -block: - _target_: text_recognizer.networks.conformer.ConformerBlock - dim: &dim 128 - attn: - _target_: text_recognizer.networks.transformer.Attention - dim: *dim - num_heads: 8 - dim_head: 64 - dropout_rate: 0.1 - causal: false - rotary_embedding: - _target_: text_recognizer.networks.transformer.RotaryEmbedding - dim: 64 - ff: - _target_: text_recognizer.networks.conformer.Feedforward - dim: *dim - expansion_factor: 4 - dropout: 0.1 - conv: - _target_: text_recognizer.networks.conformer.ConformerConv - dim: *dim - expansion_factor: 2 - kernel_size: 31 - dropout: 0.1 |