Commit
·
7a1854e
1
Parent(s):
f6c0dcc
Update hyperparams.yaml
Browse files- hyperparams.yaml +4 -1
hyperparams.yaml
CHANGED
@@ -49,6 +49,8 @@ ctc_cost: !name:speechbrain.nnet.losses.ctc_loss
|
|
49 |
model: !new:torch.nn.ModuleList
|
50 |
- [!ref <enc>, !ref <output_lin>]
|
51 |
|
|
|
|
|
52 |
encoder:
|
53 |
wav2vec2: !ref <wav2vec2>
|
54 |
enc: !ref <enc>
|
@@ -63,4 +65,5 @@ modules:
|
|
63 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
64 |
loadables:
|
65 |
wav2vec2: !ref <wav2vec2>
|
66 |
-
model: !ref <model>
|
|
|
|
49 |
model: !new:torch.nn.ModuleList
|
50 |
- [!ref <enc>, !ref <output_lin>]
|
51 |
|
52 |
+
tokenizer: !new:speechbrain.dataio.encoder.CTCTextEncoder
|
53 |
+
|
54 |
encoder:
|
55 |
wav2vec2: !ref <wav2vec2>
|
56 |
enc: !ref <enc>
|
|
|
65 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
66 |
loadables:
|
67 |
wav2vec2: !ref <wav2vec2>
|
68 |
+
model: !ref <model>
|
69 |
+
tokenizer: !ref <tokenizer>
|