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