Commit
•
ea5d8ac
1
Parent(s):
a61b505
Update hyperparams.yaml
Browse files- hyperparams.yaml +8 -5
hyperparams.yaml
CHANGED
@@ -46,16 +46,19 @@ log_softmax: !new:speechbrain.nnet.activations.Softmax
|
|
46 |
ctc_cost: !name:speechbrain.nnet.losses.ctc_loss
|
47 |
blank_index: !ref <blank_index>
|
48 |
|
49 |
-
|
|
|
|
|
|
|
50 |
wav2vec2: !ref <wav2vec2>
|
51 |
enc: !ref <enc>
|
52 |
output_lin: !ref <output_lin>
|
53 |
|
54 |
-
|
55 |
-
|
56 |
|
57 |
-
|
58 |
-
|
59 |
|
60 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
61 |
loadables:
|
|
|
46 |
ctc_cost: !name:speechbrain.nnet.losses.ctc_loss
|
47 |
blank_index: !ref <blank_index>
|
48 |
|
49 |
+
model: !new:torch.nn.ModuleList
|
50 |
+
- [!ref <enc>, !ref <output_lin>]
|
51 |
+
|
52 |
+
encoder:
|
53 |
wav2vec2: !ref <wav2vec2>
|
54 |
enc: !ref <enc>
|
55 |
output_lin: !ref <output_lin>
|
56 |
|
57 |
+
decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
|
58 |
+
blank_id: !ref <blank_index>
|
59 |
|
60 |
+
modules:
|
61 |
+
encoder: !ref <encoder>
|
62 |
|
63 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
64 |
loadables:
|