Update hyperparams.yaml
Browse files- hyperparams.yaml +2 -1
hyperparams.yaml
CHANGED
|
@@ -57,7 +57,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
|
|
| 57 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
| 58 |
activation3: !new:torch.nn.LeakyReLU
|
| 59 |
|
| 60 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
| 61 |
source: !ref <wav2vec2_hub>
|
| 62 |
output_norm: True
|
| 63 |
freeze: True
|
|
@@ -82,6 +82,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
| 82 |
wav2vec2: !ref <wav2vec2>
|
| 83 |
enc: !ref <enc>
|
| 84 |
ctc_lin: !ref <ctc_lin>
|
|
|
|
| 85 |
|
| 86 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
| 87 |
blank_id: !ref <blank_index>
|
|
|
|
| 57 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
| 58 |
activation3: !new:torch.nn.LeakyReLU
|
| 59 |
|
| 60 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
| 61 |
source: !ref <wav2vec2_hub>
|
| 62 |
output_norm: True
|
| 63 |
freeze: True
|
|
|
|
| 82 |
wav2vec2: !ref <wav2vec2>
|
| 83 |
enc: !ref <enc>
|
| 84 |
ctc_lin: !ref <ctc_lin>
|
| 85 |
+
log_softmax: !ref <log_softmax>
|
| 86 |
|
| 87 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
| 88 |
blank_id: !ref <blank_index>
|