Commit
·
d124261
1
Parent(s):
876b26c
add beam search to modules
Browse files- hyperparams.yaml +7 -7
hyperparams.yaml
CHANGED
@@ -28,7 +28,7 @@ temperature: 1.25
|
|
28 |
# Models
|
29 |
asr_model: !apply:speechbrain.pretrained.EncoderDecoderASR.from_hparams
|
30 |
source: speechbrain/asr-crdnn-rnnlm-librispeech
|
31 |
-
run_opts: {"device":"cuda:0"}
|
32 |
|
33 |
slu_enc: !new:speechbrain.nnet.containers.Sequential
|
34 |
input_shape: [null, null, !ref <ASR_encoder_dim>]
|
@@ -60,12 +60,6 @@ seq_lin: !new:speechbrain.nnet.linear.Linear
|
|
60 |
input_size: !ref <dec_neurons>
|
61 |
n_neurons: !ref <output_neurons>
|
62 |
|
63 |
-
modules:
|
64 |
-
slu_enc: !ref <slu_enc>
|
65 |
-
output_emb: !ref <output_emb>
|
66 |
-
dec: !ref <dec>
|
67 |
-
seq_lin: !ref <seq_lin>
|
68 |
-
|
69 |
model: !new:torch.nn.ModuleList
|
70 |
- [!ref <slu_enc>, !ref <output_emb>,
|
71 |
!ref <dec>, !ref <seq_lin>]
|
@@ -94,4 +88,10 @@ beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
|
94 |
log_softmax: !new:speechbrain.nnet.activations.Softmax
|
95 |
apply_log: True
|
96 |
|
|
|
|
|
|
|
|
|
|
|
|
|
97 |
|
|
|
28 |
# Models
|
29 |
asr_model: !apply:speechbrain.pretrained.EncoderDecoderASR.from_hparams
|
30 |
source: speechbrain/asr-crdnn-rnnlm-librispeech
|
31 |
+
#run_opts: {"device":"cuda:0"}
|
32 |
|
33 |
slu_enc: !new:speechbrain.nnet.containers.Sequential
|
34 |
input_shape: [null, null, !ref <ASR_encoder_dim>]
|
|
|
60 |
input_size: !ref <dec_neurons>
|
61 |
n_neurons: !ref <output_neurons>
|
62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
model: !new:torch.nn.ModuleList
|
64 |
- [!ref <slu_enc>, !ref <output_emb>,
|
65 |
!ref <dec>, !ref <seq_lin>]
|
|
|
88 |
log_softmax: !new:speechbrain.nnet.activations.Softmax
|
89 |
apply_log: True
|
90 |
|
91 |
+
modules:
|
92 |
+
slu_enc: !ref <slu_enc>
|
93 |
+
output_emb: !ref <output_emb>
|
94 |
+
dec: !ref <dec>
|
95 |
+
seq_lin: !ref <seq_lin>
|
96 |
+
beam_searcher: !ref <beam_searcher>
|
97 |
|