Commit
•
0157ee1
1
Parent(s):
27673c0
print preds for debugging
Browse files
run_speech_recognition_seq2seq_streaming.py
CHANGED
@@ -812,6 +812,7 @@ def main():
|
|
812 |
num_beams=training_args.generation_num_beams,
|
813 |
)
|
814 |
logger.info("*** Test prediction done ***")
|
|
|
815 |
predictions = processor.batch_decode(predictions.predictions)
|
816 |
labels = processor.batch_decode(predictions.label_ids)
|
817 |
pred_labels = [f"Prediction: {pred}\nLabel: {label}\n" for pred, label in zip(predictions, labels)]
|
|
|
812 |
num_beams=training_args.generation_num_beams,
|
813 |
)
|
814 |
logger.info("*** Test prediction done ***")
|
815 |
+
print(predictions)
|
816 |
predictions = processor.batch_decode(predictions.predictions)
|
817 |
labels = processor.batch_decode(predictions.label_ids)
|
818 |
pred_labels = [f"Prediction: {pred}\nLabel: {label}\n" for pred, label in zip(predictions, labels)]
|