Adel-Moumen
commited on
Commit
•
269c16b
1
Parent(s):
a83e454
Update hyperparams.yaml
Browse files- hyperparams.yaml +2 -1
hyperparams.yaml
CHANGED
@@ -58,7 +58,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
|
|
58 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
59 |
activation3: !new:torch.nn.LeakyReLU
|
60 |
|
61 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
62 |
source: !ref <wav2vec2_hub>
|
63 |
output_norm: True
|
64 |
freeze: True
|
@@ -83,6 +83,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
83 |
wav2vec2: !ref <wav2vec2>
|
84 |
enc: !ref <enc>
|
85 |
ctc_lin: !ref <ctc_lin>
|
|
|
86 |
|
87 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
88 |
blank_id: !ref <blank_index>
|
|
|
58 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
59 |
activation3: !new:torch.nn.LeakyReLU
|
60 |
|
61 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
62 |
source: !ref <wav2vec2_hub>
|
63 |
output_norm: True
|
64 |
freeze: True
|
|
|
83 |
wav2vec2: !ref <wav2vec2>
|
84 |
enc: !ref <enc>
|
85 |
ctc_lin: !ref <ctc_lin>
|
86 |
+
log_softmax: !ref <log_softmax>
|
87 |
|
88 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
89 |
blank_id: !ref <blank_index>
|