Update hyperparams.yaml

hyperparams.yaml

@@ -4,11 +4,11 @@ HPARAMS_NEEDED: ["wav2vec_output_dim", "emb_size", "dec_neurons", "dec_layers",
 MODULES_NEEDED: ["encoder_w2v2", "embedding", "ctc_lin", "seq_lin"]
 
 # Pretrain folder (HuggingFace)
-output_folder: !ref output_folder_seq2seq_cv_podcast_arhiv_augmentation_128_emb_5000_vocab
+# output_folder: !ref output_folder_seq2seq_cv_podcast_arhiv_augmentation_128_emb_5000_vocab
 pretrained_path: Porjaz/wav2vec2-aed-macedonian-asr
 wav2vec2_hub: facebook/wav2vec2-large-xlsr-53
-save_folder: !ref <output_folder>/save
-wav2vec2_folder: !ref <save_folder>/wav2vec2_checkpoint
+# save_folder: !ref <output_folder>/save
+# wav2vec2_folder: !ref <save_folder>/wav2vec2_checkpoint
 
 ####################### Training Parameters ####################################
 
@@ -25,22 +25,20 @@ bos_index: 0
 eos_index: 0
 unk_index: 0
 
+# Decoding parameters
 # Decoding parameters
 min_decode_ratio: 0.0
 max_decode_ratio: 1.0
 valid_beam_size: 10
-test_beam_size: 10
+test_beam_size: 20
 using_eos_threshold: True
 eos_threshold: 1.5
 using_max_attn_shift: True
-max_attn_shift: 300
+max_attn_shift: 700
+length_normalization: True
 temperature: 1.0
-ctc_window_size: 200
-temperature_lm: 1.25
 # Scoring parameters
-ctc_weight_decode: 0.0
 coverage_penalty: 1.5
-lm_weight: 0.0
 
 
 # Wav2vec2 encoder
@@ -109,7 +107,7 @@ test_search: !new:speechbrain.decoders.S2SRNNBeamSearcher
     using_max_attn_shift: !ref <using_max_attn_shift>
     max_attn_shift: !ref <max_attn_shift>
     temperature: !ref <temperature>
-      #scorer: !ref <scorer>
+    scorer: !ref <scorer>
 
 
 ############################## Logging and Pretrainer ##########################