marinone94 commited on
Commit
8ce7989
1 Parent(s): a057c82
run_speech_recognition_seq2seq_streaming.py CHANGED
@@ -780,7 +780,7 @@ def main():
780
 
781
  # Run a test prediction to check outputs
782
  predictions = trainer.predict(
783
- test_dataset=vectorized_datasets["test"].shuffle(seed=training_args.seed).select(range(5)),
784
  metric_key_prefix="test",
785
  max_length=training_args.generation_max_length,
786
  num_beams=training_args.generation_num_beams,
 
780
 
781
  # Run a test prediction to check outputs
782
  predictions = trainer.predict(
783
+ test_dataset=vectorized_datasets["eval"].shuffle(seed=training_args.seed).select(range(5)),
784
  metric_key_prefix="test",
785
  max_length=training_args.generation_max_length,
786
  num_beams=training_args.generation_num_beams,