fix for GPU inference
Browse files- 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:
|