File size: 378 Bytes
a534790 2897817 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | model:
name: "metro-small"
d_model: 384
n_layers: 12
n_heads: 6
ff_multiplier: 3
conv_kernel_size: 31
dropout: 0.1
conv_expansion_factor: 2
se_ratio: 4
stochastic_depth_rate: 0.05
audio:
sample_rate: 16000
n_mels: 80
n_fft: 512
hop_length: 160
win_length: 400
tokenizer:
type: "bpe"
vocab_size: 5000
training:
intermediate_ctc_layers: [6]
|