Adel-Moumen
commited on
Commit
•
356e347
1
Parent(s):
9035a95
Update hyperparams.yaml
Browse files- hyperparams.yaml +3 -2
hyperparams.yaml
CHANGED
@@ -28,7 +28,7 @@ enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
|
|
28 |
dnn_blocks: !ref <dnn_layers>
|
29 |
dnn_neurons: !ref <dnn_neurons>
|
30 |
|
31 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
32 |
source: !ref <wav2vec2_hub>
|
33 |
output_norm: True
|
34 |
freeze: True
|
@@ -53,6 +53,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
53 |
wav2vec2: !ref <wav2vec2>
|
54 |
enc: !ref <enc>
|
55 |
ctc_lin: !ref <ctc_lin>
|
|
|
56 |
|
57 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
58 |
blank_id: !ref <blank_index>
|
@@ -64,4 +65,4 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
|
64 |
loadables:
|
65 |
wav2vec2: !ref <wav2vec2>
|
66 |
asr: !ref <asr_model>
|
67 |
-
tokenizer: !ref <tokenizer>
|
|
|
28 |
dnn_blocks: !ref <dnn_layers>
|
29 |
dnn_neurons: !ref <dnn_neurons>
|
30 |
|
31 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.Wav2Vec2
|
32 |
source: !ref <wav2vec2_hub>
|
33 |
output_norm: True
|
34 |
freeze: True
|
|
|
53 |
wav2vec2: !ref <wav2vec2>
|
54 |
enc: !ref <enc>
|
55 |
ctc_lin: !ref <ctc_lin>
|
56 |
+
log_softmax: !ref <log_softmax>
|
57 |
|
58 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
59 |
blank_id: !ref <blank_index>
|
|
|
65 |
loadables:
|
66 |
wav2vec2: !ref <wav2vec2>
|
67 |
asr: !ref <asr_model>
|
68 |
+
tokenizer: !ref <tokenizer>
|