summaryrefslogtreecommitdiff
path: root/training/conf/network
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-12 23:16:20 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-12 23:16:20 +0200
commit8bb76745e43c6b4967c8e91ebaf4c4295d0b8d0b (patch)
tree5ff05d9fea92f7e5bd313d8cdc9559ccbc89a97a /training/conf/network
parent8fe4b36bf22281c84c4afee811b3435f3b50686d (diff)
Remove conformer
Diffstat (limited to 'training/conf/network')
-rw-r--r--training/conf/network/conformer.yaml36
1 files changed, 0 insertions, 36 deletions
diff --git a/training/conf/network/conformer.yaml b/training/conf/network/conformer.yaml
deleted file mode 100644
index 1d72dd5..0000000
--- a/training/conf/network/conformer.yaml
+++ /dev/null
@@ -1,36 +0,0 @@
-_target_: text_recognizer.networks.conformer.Conformer
-depth: 8
-num_classes: 57
-dim: &dim 144
-dim_gru: 144
-block:
- _target_: text_recognizer.networks.conformer.ConformerBlock
- dim: *dim
- attn:
- _target_: text_recognizer.networks.conformer.Attention
- dim: *dim
- heads: 8
- dim_head: 64
- mult: 4
- 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
-subsampler:
- _target_: text_recognizer.networks.conformer.Subsampler
- pixel_pos_embedding:
- _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding
- dim: 64
- shape: [6, 127]
- channels: 64
- height: 6
- dim: *dim
- depth: 3
- dropout: 0.1