Adel-Moumen commited on
Commit
e9d297b
1 Parent(s): 9b2cedf

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +9 -0
hyperparams.yaml CHANGED
@@ -79,23 +79,30 @@ dec: !new:speechbrain.nnet.RNN.AttentionalRNNDecoder
79
  kernel_size: 100
80
  re_init: True
81
  dropout: !ref <dropout>
 
82
  ctc_lin: !new:speechbrain.nnet.linear.Linear
83
  input_size: !ref <dnn_neurons>
84
  n_neurons: !ref <output_neurons>
 
85
  seq_lin: !new:speechbrain.nnet.linear.Linear
86
  input_size: !ref <dec_neurons>
87
  n_neurons: !ref <output_neurons>
 
88
  log_softmax: !new:speechbrain.nnet.activations.Softmax
89
  apply_log: True
 
90
  asr_model: !new:torch.nn.ModuleList
91
  - [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
 
92
  tokenizer: !new:sentencepiece.SentencePieceProcessor
 
93
  # We compose the inference (encoder) pipeline.
94
  encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
95
  input_shape: [null, null, !ref <n_mels>]
96
  compute_features: !ref <compute_features>
97
  normalize: !ref <normalizer>
98
  model: !ref <enc>
 
99
  decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
100
  embedding: !ref <emb>
101
  decoder: !ref <dec>
@@ -109,10 +116,12 @@ decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
109
  using_max_attn_shift: !ref <using_max_attn_shift>
110
  max_attn_shift: !ref <max_attn_shift>
111
  temperature: !ref <temperature>
 
112
  modules:
113
  normalizer: !ref <normalizer>
114
  encoder: !ref <encoder>
115
  decoder: !ref <decoder>
 
116
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
117
  loadables:
118
  normalizer: !ref <normalizer>
 
79
  kernel_size: 100
80
  re_init: True
81
  dropout: !ref <dropout>
82
+
83
  ctc_lin: !new:speechbrain.nnet.linear.Linear
84
  input_size: !ref <dnn_neurons>
85
  n_neurons: !ref <output_neurons>
86
+
87
  seq_lin: !new:speechbrain.nnet.linear.Linear
88
  input_size: !ref <dec_neurons>
89
  n_neurons: !ref <output_neurons>
90
+
91
  log_softmax: !new:speechbrain.nnet.activations.Softmax
92
  apply_log: True
93
+
94
  asr_model: !new:torch.nn.ModuleList
95
  - [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
96
+
97
  tokenizer: !new:sentencepiece.SentencePieceProcessor
98
+
99
  # We compose the inference (encoder) pipeline.
100
  encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
101
  input_shape: [null, null, !ref <n_mels>]
102
  compute_features: !ref <compute_features>
103
  normalize: !ref <normalizer>
104
  model: !ref <enc>
105
+
106
  decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
107
  embedding: !ref <emb>
108
  decoder: !ref <dec>
 
116
  using_max_attn_shift: !ref <using_max_attn_shift>
117
  max_attn_shift: !ref <max_attn_shift>
118
  temperature: !ref <temperature>
119
+
120
  modules:
121
  normalizer: !ref <normalizer>
122
  encoder: !ref <encoder>
123
  decoder: !ref <decoder>
124
+
125
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
126
  loadables:
127
  normalizer: !ref <normalizer>