minor update
Browse files- hyperparams.yaml +2 -2
hyperparams.yaml
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
pretrained_path: HaNguyen/IWSLT-ast-w2v2-
|
2 |
|
3 |
lang: fr #for the BLEU score detokenization
|
4 |
target_lang: fr_XX # for mbart initialization
|
@@ -40,7 +40,7 @@ test_beam_size: 5
|
|
40 |
|
41 |
############################## models ################################
|
42 |
#wav2vec model
|
43 |
-
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.
|
44 |
source: !ref <wav2vec2_hub>
|
45 |
output_norm: True
|
46 |
freeze: !ref <wav2vec2_frozen>
|
|
|
1 |
+
pretrained_path: HaNguyen/IWSLT-ast-w2v2-mbart
|
2 |
|
3 |
lang: fr #for the BLEU score detokenization
|
4 |
target_lang: fr_XX # for mbart initialization
|
|
|
40 |
|
41 |
############################## models ################################
|
42 |
#wav2vec model
|
43 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
44 |
source: !ref <wav2vec2_hub>
|
45 |
output_norm: True
|
46 |
freeze: !ref <wav2vec2_frozen>
|