GaelleLaperriere commited on
Commit
ea5d8ac
1 Parent(s): a61b505

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. 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
- modules:
 
 
 
50
  wav2vec2: !ref <wav2vec2>
51
  enc: !ref <enc>
52
  output_lin: !ref <output_lin>
53
 
54
- model: !new:torch.nn.ModuleList
55
- - [!ref <enc>, !ref <output_lin>]
56
 
57
- model_wav2vec2: !new:torch.nn.ModuleList
58
- - [!ref <wav2vec2>]
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: