marinone94
commited on
Commit
•
8ce7989
1
Parent(s):
a057c82
fix typo
Browse files
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["
|
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,
|