chaanks commited on
Commit
fbeb418
1 Parent(s): 3dab7eb

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +5 -4
hyperparams.yaml CHANGED
@@ -28,9 +28,6 @@ test_beam_size: 8
28
  # Model parameters
29
  freeze_whisper: True
30
  freeze_encoder: True
31
-
32
- # tokenizer: !name:transformers.models.whisper.tokenization_whisper.WhisperTokenizer.from_pretrained
33
- # pretrained_model_name_or_path: !ref <whisper_hub>
34
 
35
 
36
  whisper: !new:speechbrain.lobes.models.huggingface_whisper.HuggingFaceWhisper
@@ -60,4 +57,8 @@ decoder: !new:speechbrain.decoders.seq2seq.S2SWhisperGreedySearch
60
 
61
  modules:
62
  whisper: !ref <whisper>
63
- decoder: !ref <decoder>
 
 
 
 
 
28
  # Model parameters
29
  freeze_whisper: True
30
  freeze_encoder: True
 
 
 
31
 
32
 
33
  whisper: !new:speechbrain.lobes.models.huggingface_whisper.HuggingFaceWhisper
 
57
 
58
  modules:
59
  whisper: !ref <whisper>
60
+ decoder: !ref <decoder>
61
+
62
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
63
+ loadables:
64
+ whisper: !ref <whisper>