Adel-Moumen commited on
Commit
4a3b316
1 Parent(s): 4c78f98

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +12 -3
hyperparams.yaml CHANGED
@@ -76,19 +76,28 @@ seq_lin: !new:speechbrain.nnet.linear.Linear
76
  input_size: !ref <d_model>
77
  n_neurons: !ref <output_neurons>
78
 
 
 
 
 
 
 
 
 
 
 
79
  decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
80
- modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
81
  bos_index: !ref <bos_index>
82
  eos_index: !ref <eos_index>
83
- blank_index: !ref <blank_index>
84
  min_decode_ratio: !ref <min_decode_ratio>
85
  max_decode_ratio: !ref <max_decode_ratio>
86
  beam_size: !ref <beam_size>
87
- ctc_weight: !ref <ctc_weight_decode>
88
  temperature: 1.15
89
  temperature_lm: 1.15
90
  using_eos_threshold: False
91
  length_normalization: True
 
92
 
93
  log_softmax: !new:torch.nn.LogSoftmax
94
  dim: -1
 
76
  input_size: !ref <d_model>
77
  n_neurons: !ref <output_neurons>
78
 
79
+ ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
80
+ eos_index: !ref <eos_index>
81
+ blank_index: !ref <blank_index>
82
+ ctc_fc: !ref <ctc_lin>
83
+
84
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
85
+ full_scorers: [!ref <ctc_scorer>]
86
+ weights:
87
+ ctc: !ref <ctc_weight_decode>
88
+
89
  decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
90
+ modules: [!ref <Transformer>, !ref <seq_lin>]
91
  bos_index: !ref <bos_index>
92
  eos_index: !ref <eos_index>
 
93
  min_decode_ratio: !ref <min_decode_ratio>
94
  max_decode_ratio: !ref <max_decode_ratio>
95
  beam_size: !ref <beam_size>
 
96
  temperature: 1.15
97
  temperature_lm: 1.15
98
  using_eos_threshold: False
99
  length_normalization: True
100
+ scorer: !ref <scorer>
101
 
102
  log_softmax: !new:torch.nn.LogSoftmax
103
  dim: -1