marinone94 commited on
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
- 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)]
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
- predictions = processor.batch_decode(predictions.predictions)
823
  labels = processor.batch_decode(predictions.label_ids)
824
- pred_labels = [f"Prediction: {pred}\nLabel: {label}\n" for pred, label in zip(predictions, labels)]
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