subakany commited on
Commit
c429442
1 Parent(s): e915e01

added pretrainer to the hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +7 -0
hyperparams.yaml CHANGED
@@ -173,3 +173,10 @@ checkpointer: !new:speechbrain.utils.checkpoints.Checkpointer
173
  lr_scheduler: *id007
174
  train_logger: !new:speechbrain.utils.train_logger.FileTrainLogger
175
  save_file: results/sepformer-wham-enhancement/1234/train_log.txt
 
 
 
 
 
 
 
 
173
  lr_scheduler: *id007
174
  train_logger: !new:speechbrain.utils.train_logger.FileTrainLogger
175
  save_file: results/sepformer-wham-enhancement/1234/train_log.txt
176
+
177
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
178
+ loadables:
179
+ encoder: !ref <Encoder>
180
+ masknet: !ref <MaskNet>
181
+ decoder: !ref <Decoder>
182
+