summaryrefslogtreecommitdiff
path: root/training/conf/network/conformer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-09 22:36:04 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-09 22:36:04 +0200
commit5c28ad1905f63838afbc7ee6c5ec31ff1310e3a1 (patch)
tree594796cc7d8bd01fedb167208dbf2593f2b898a1 /training/conf/network/conformer
parent4c7f720fe42278ec27e87bb1c3d5fcc6f1f822c2 (diff)
Move conformer config
Diffstat (limited to 'training/conf/network/conformer')
-rw-r--r--training/conf/network/conformer/conformer.yaml27
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