Adel-Moumen
commited on
Update hyperparams.yaml
Browse files- hyperparams.yaml +2 -4
hyperparams.yaml
CHANGED
@@ -41,7 +41,7 @@ enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
|
|
41 |
dnn_blocks: !ref <dnn_layers>
|
42 |
dnn_neurons: !ref <dnn_neurons>
|
43 |
|
44 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
45 |
source: !ref <wav2vec2_hub>
|
46 |
output_norm: True
|
47 |
freeze: True
|
@@ -90,15 +90,13 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
|
|
90 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
91 |
wav2vec2: !ref <wav2vec2>
|
92 |
enc: !ref <enc>
|
93 |
-
|
94 |
decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
95 |
embedding: !ref <emb>
|
96 |
decoder: !ref <dec>
|
97 |
linear: !ref <seq_lin>
|
98 |
-
ctc_linear: !ref <ctc_lin>
|
99 |
bos_index: !ref <bos_index>
|
100 |
eos_index: !ref <eos_index>
|
101 |
-
blank_index: !ref <blank_index>
|
102 |
min_decode_ratio: !ref <min_decode_ratio>
|
103 |
max_decode_ratio: !ref <max_decode_ratio>
|
104 |
beam_size: !ref <beam_size>
|
|
|
41 |
dnn_blocks: !ref <dnn_layers>
|
42 |
dnn_neurons: !ref <dnn_neurons>
|
43 |
|
44 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
45 |
source: !ref <wav2vec2_hub>
|
46 |
output_norm: True
|
47 |
freeze: True
|
|
|
90 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
91 |
wav2vec2: !ref <wav2vec2>
|
92 |
enc: !ref <enc>
|
93 |
+
|
94 |
decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
95 |
embedding: !ref <emb>
|
96 |
decoder: !ref <dec>
|
97 |
linear: !ref <seq_lin>
|
|
|
98 |
bos_index: !ref <bos_index>
|
99 |
eos_index: !ref <eos_index>
|
|
|
100 |
min_decode_ratio: !ref <min_decode_ratio>
|
101 |
max_decode_ratio: !ref <max_decode_ratio>
|
102 |
beam_size: !ref <beam_size>
|