Commit
•
d9579c1
1
Parent(s):
b0ee2eb
Update hyperparams.yaml
Browse files- hyperparams.yaml +4 -2
hyperparams.yaml
CHANGED
|
@@ -5,7 +5,7 @@
|
|
| 5 |
# ################################
|
| 6 |
|
| 7 |
sample_rate: 16000
|
| 8 |
-
wav2vec2_hub: facebook/wav2vec2-large-
|
| 9 |
|
| 10 |
# BPE parameters
|
| 11 |
token_type: unigram # ["unigram", "bpe", "char"]
|
|
@@ -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
|
|
@@ -95,8 +95,10 @@ 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>
|
|
|
|
| 5 |
# ################################
|
| 6 |
|
| 7 |
sample_rate: 16000
|
| 8 |
+
wav2vec2_hub: facebook/wav2vec2-large-xlsr-53
|
| 9 |
|
| 10 |
# BPE parameters
|
| 11 |
token_type: unigram # ["unigram", "bpe", "char"]
|
|
|
|
| 41 |
dnn_blocks: !ref <dnn_layers>
|
| 42 |
dnn_neurons: !ref <dnn_neurons>
|
| 43 |
|
| 44 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.wav2vec2.Wav2Vec2
|
| 45 |
source: !ref <wav2vec2_hub>
|
| 46 |
output_norm: True
|
| 47 |
freeze: True
|
|
|
|
| 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>
|