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:
|