subakany commited on
Commit
5db677b
1 Parent(s): 0a8f294

added pretrainer class to hyperparams.yaml

Browse files
Files changed (1) hide show
  1. 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>