Commit
•
b9bdaf4
1
Parent(s):
0157ee1
ifx overwriting var
Browse files
run_speech_recognition_seq2seq_streaming.py
CHANGED
@@ -812,16 +812,15 @@ def main():
|
|
812 |
num_beams=training_args.generation_num_beams,
|
813 |
)
|
814 |
logger.info("*** Test prediction done ***")
|
815 |
-
|
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(
|
819 |
logger.info("Before setting language and task")
|
820 |
logger.info(f"{pred_labels}")
|
821 |
trainer.data_collator.processor.tokenizer.set_prefix_tokens(language=data_args.language_eval, task=data_args.task)
|
822 |
-
|
823 |
labels = processor.batch_decode(predictions.label_ids)
|
824 |
-
pred_labels = [f"Prediction: {pred}\nLabel: {label}\n" for pred, label in zip(
|
825 |
logger.info("After setting language and task")
|
826 |
logger.info(f"{pred_labels}")
|
827 |
|
|
|
812 |
num_beams=training_args.generation_num_beams,
|
813 |
)
|
814 |
logger.info("*** Test prediction done ***")
|
815 |
+
preds = 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(preds, labels)]
|
818 |
logger.info("Before setting language and task")
|
819 |
logger.info(f"{pred_labels}")
|
820 |
trainer.data_collator.processor.tokenizer.set_prefix_tokens(language=data_args.language_eval, task=data_args.task)
|
821 |
+
preds = processor.batch_decode(predictions.predictions)
|
822 |
labels = processor.batch_decode(predictions.label_ids)
|
823 |
+
pred_labels = [f"Prediction: {pred}\nLabel: {label}\n" for pred, label in zip(preds, labels)]
|
824 |
logger.info("After setting language and task")
|
825 |
logger.info(f"{pred_labels}")
|
826 |
|