Update hyperparams.yaml
Browse files- hyperparams.yaml +5 -2
hyperparams.yaml
CHANGED
|
@@ -14,7 +14,9 @@ feats_dim: 1024
|
|
| 14 |
activation: !name:torch.nn.LeakyReLU
|
| 15 |
dnn_blocks: 3
|
| 16 |
dnn_neurons: 512
|
| 17 |
-
|
|
|
|
|
|
|
| 18 |
# Decoding parameters:
|
| 19 |
blank_index: 0
|
| 20 |
|
|
@@ -23,7 +25,7 @@ output_neurons: 67
|
|
| 23 |
|
| 24 |
# ------ Functions and classes
|
| 25 |
|
| 26 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
| 27 |
source: !ref <wav2vec_url>
|
| 28 |
output_norm: True
|
| 29 |
freeze: Freeze
|
|
@@ -49,6 +51,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
| 49 |
wav2vec2: !ref <wav2vec2>
|
| 50 |
enc: !ref <enc>
|
| 51 |
output_lin: !ref <output_lin>
|
|
|
|
| 52 |
|
| 53 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
| 54 |
blank_id: !ref <blank_index>
|
|
|
|
| 14 |
activation: !name:torch.nn.LeakyReLU
|
| 15 |
dnn_blocks: 3
|
| 16 |
dnn_neurons: 512
|
| 17 |
+
log_softmax: !new:torch.nn.LogSoftmax
|
| 18 |
+
dim: -1
|
| 19 |
+
|
| 20 |
# Decoding parameters:
|
| 21 |
blank_index: 0
|
| 22 |
|
|
|
|
| 25 |
|
| 26 |
# ------ Functions and classes
|
| 27 |
|
| 28 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
| 29 |
source: !ref <wav2vec_url>
|
| 30 |
output_norm: True
|
| 31 |
freeze: Freeze
|
|
|
|
| 51 |
wav2vec2: !ref <wav2vec2>
|
| 52 |
enc: !ref <enc>
|
| 53 |
output_lin: !ref <output_lin>
|
| 54 |
+
log_softmax: !ref <log_softmax>
|
| 55 |
|
| 56 |
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
| 57 |
blank_id: !ref <blank_index>
|