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>
|