marinone94
commited on
Commit
·
ace0e10
1
Parent(s):
f3dac86
fix print len eval
Browse files
run_speech_recognition_ctc.py
CHANGED
@@ -815,8 +815,8 @@ def main():
|
|
815 |
# we do not want to group tokens when computing the metrics
|
816 |
label_str = tokenizer.batch_decode(pred.label_ids, group_tokens=False)
|
817 |
|
818 |
-
print(pred_str)
|
819 |
-
print(label_str)
|
820 |
|
821 |
metrics = {k: v.compute(predictions=pred_str, references=label_str) for k, v in eval_metrics.items()}
|
822 |
|
|
|
815 |
# we do not want to group tokens when computing the metrics
|
816 |
label_str = tokenizer.batch_decode(pred.label_ids, group_tokens=False)
|
817 |
|
818 |
+
print(pred_str[:10])
|
819 |
+
print(label_str[:10])
|
820 |
|
821 |
metrics = {k: v.compute(predictions=pred_str, references=label_str) for k, v in eval_metrics.items()}
|
822 |
|