marinone94 commited on
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)]