set max length / num beams according to gen kwargs

huggingface · Oct 13, 2022 · ecd4e03 · ecd4e03
1 parent f398822
commit ecd4e03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py
@@ -514,7 +514,7 @@ def compute_metrics(pred):
     if training_args.do_eval:
         logger.info("*** Evaluate ***")
         metrics = trainer.evaluate(
-            metric_key_prefix="eval", max_length=model.config.max_length, num_beams=model.config.num_beams
+            metric_key_prefix="eval", max_length=training_args.generation_max_length, num_beams=training_args.generation_num_beams
         )
         max_eval_samples = (
             data_args.max_eval_samples if data_args.max_eval_samples is not None else len(vectorized_datasets["eval"])