Titouan commited on
Commit
234e94e
·
verified ·
1 Parent(s): 2996a98

fix for GPU inference

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -0
hyperparams.yaml CHANGED
@@ -119,6 +119,8 @@ modules:
119
  encoder: !ref <encoder>
120
  transformer: !ref <Transformer>
121
  decoder: !ref <decoder>
 
 
122
 
123
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
124
  loadables:
 
119
  encoder: !ref <encoder>
120
  transformer: !ref <Transformer>
121
  decoder: !ref <decoder>
122
+ ctc_lin: !ref <ctc_lin>
123
+ seq_ln: !ref <seq_lin>
124
 
125
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
126
  loadables: