Commit
·
059ce0a
1
Parent(s):
33fa953
log eval
Browse files
run_speech_recognition_ctc.py
CHANGED
|
@@ -814,6 +814,9 @@ def main():
|
|
| 814 |
# we do not want to group tokens when computing the metrics
|
| 815 |
label_str = tokenizer.batch_decode(pred.label_ids, group_tokens=False)
|
| 816 |
|
|
|
|
|
|
|
|
|
|
| 817 |
metrics = {k: v.compute(predictions=pred_str, references=label_str) for k, v in eval_metrics.items()}
|
| 818 |
|
| 819 |
return metrics
|
|
|
|
| 814 |
# we do not want to group tokens when computing the metrics
|
| 815 |
label_str = tokenizer.batch_decode(pred.label_ids, group_tokens=False)
|
| 816 |
|
| 817 |
+
print(pred_str)
|
| 818 |
+
print(label_str)
|
| 819 |
+
|
| 820 |
metrics = {k: v.compute(predictions=pred_str, references=label_str) for k, v in eval_metrics.items()}
|
| 821 |
|
| 822 |
return metrics
|