subakany commited on
Commit ·
5db677b
1
Parent(s): 0a8f294
added pretrainer class to hyperparams.yaml
Browse files- hyperparams.yaml +6 -0
hyperparams.yaml
CHANGED
|
@@ -172,3 +172,9 @@ checkpointer: !new:speechbrain.utils.checkpoints.Checkpointer
|
|
| 172 |
lr_scheduler: *id007
|
| 173 |
train_logger: !new:speechbrain.utils.train_logger.FileTrainLogger
|
| 174 |
save_file: results/sepformer-whamr/1234/train_log.txt
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 172 |
lr_scheduler: *id007
|
| 173 |
train_logger: !new:speechbrain.utils.train_logger.FileTrainLogger
|
| 174 |
save_file: results/sepformer-whamr/1234/train_log.txt
|
| 175 |
+
|
| 176 |
+
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
| 177 |
+
loadables:
|
| 178 |
+
masknet: !ref <MaskNet>
|
| 179 |
+
encoder: !ref <Encoder>
|
| 180 |
+
decoder: !ref <Decoder>
|