forked from ictnlp/StreamSpeech
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_mtl_asr_st_ctcst.yaml
36 lines (36 loc) · 1.18 KB
/
config_mtl_asr_st_ctcst.yaml
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
27
28
29
30
31
32
33
34
35
36
target_unigram:
decoder_type: transformer
dict: /data/zhangshaolei/StreamSpeech/configs/fr-en/tgt_unigram6000/spm_unigram_fr.txt
data: /data/zhangshaolei/StreamSpeech/configs/fr-en/tgt_unigram6000
loss_weight: 8.0
rdrop_alpha: 0.0
decoder_args:
decoder_layers: 4
decoder_embed_dim: 512
decoder_ffn_embed_dim: 2048
decoder_attention_heads: 8
label_smoothing: 0.1
source_unigram:
decoder_type: ctc
dict: /data/zhangshaolei/StreamSpeech/configs/fr-en/src_unigram6000/spm_unigram_fr.txt
data: /data/zhangshaolei/StreamSpeech/configs/fr-en/src_unigram6000
loss_weight: 4.0
rdrop_alpha: 0.0
decoder_args:
decoder_layers: 0
decoder_embed_dim: 512
decoder_ffn_embed_dim: 2048
decoder_attention_heads: 8
label_smoothing: 0.1
ctc_target_unigram:
decoder_type: ctc
dict: /data/zhangshaolei/StreamSpeech/configs/fr-en/tgt_unigram6000/spm_unigram_fr.txt
data: /data/zhangshaolei/StreamSpeech/configs/fr-en/tgt_unigram6000
loss_weight: 4.0
rdrop_alpha: 0.0
decoder_args:
decoder_layers: 0
decoder_embed_dim: 512
decoder_ffn_embed_dim: 2048
decoder_attention_heads: 8
label_smoothing: 0.1