Adel-Moumen commited on
Commit
1edc1a1
1 Parent(s): e7f9f8b

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +13 -4
hyperparams.yaml CHANGED
@@ -86,17 +86,26 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
86
  wav2vec2: !ref <wav2vec2>
87
  transformer_encoder: !ref <Tencoder>
88
 
89
- decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
90
- modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
91
- bos_index: !ref <bos_index>
92
  eos_index: !ref <eos_index>
93
  blank_index: !ref <blank_index>
 
 
 
 
 
 
 
 
 
 
 
94
  min_decode_ratio: !ref <min_decode_ratio>
95
  max_decode_ratio: !ref <max_decode_ratio>
96
  beam_size: !ref <test_beam_size>
97
- ctc_weight: !ref <ctc_weight_decode>
98
  using_eos_threshold: False
99
  length_normalization: True
 
100
 
101
  modules:
102
  encoder: !ref <encoder>
 
86
  wav2vec2: !ref <wav2vec2>
87
  transformer_encoder: !ref <Tencoder>
88
 
89
+ ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
 
 
90
  eos_index: !ref <eos_index>
91
  blank_index: !ref <blank_index>
92
+ ctc_fc: !ref <ctc_lin>
93
+
94
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
95
+ full_scorers: [!ref <ctc_scorer>]
96
+ weights:
97
+ ctc: !ref <ctc_weight_decode>
98
+
99
+ decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
100
+ modules: [!ref <Transformer>, !ref <seq_lin>]
101
+ bos_index: !ref <bos_index>
102
+ eos_index: !ref <eos_index>
103
  min_decode_ratio: !ref <min_decode_ratio>
104
  max_decode_ratio: !ref <max_decode_ratio>
105
  beam_size: !ref <test_beam_size>
 
106
  using_eos_threshold: False
107
  length_normalization: True
108
+ scorer: !ref <scorer>
109
 
110
  modules:
111
  encoder: !ref <encoder>