lorenlugosch commited on
Commit
d124261
1 Parent(s): 876b26c

add beam search to modules

Browse files
Files changed (1) hide show
  1. 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