Update hyperparams.yaml
Browse files- 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>
|