marinone94
commited on
Commit
•
5920347
1
Parent(s):
8ce7989
use take instead of select
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["eval"].shuffle(seed=training_args.seed).
|
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).take(5),
|
784 |
metric_key_prefix="test",
|
785 |
max_length=training_args.generation_max_length,
|
786 |
num_beams=training_args.generation_num_beams,
|