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>
|