Skip to content
Snippets Groups Projects
Commit 5641f4c8 authored by Mélodie Boillet's avatar Mélodie Boillet Committed by Yoann Schneider
Browse files

Fix CER/WER always equal to 1.0

parent 6cdfc750
No related branches found
No related tags found
1 merge request!33Fix CER/WER always equal to 1.0
......@@ -1262,7 +1262,7 @@ class Manager(OCRManager):
)
predicted_tokens_len += 1
prediction_len[reached_end is False] = i + 1
prediction_len[reached_end == False] = i + 1 # noqa E712
if torch.all(reached_end):
break
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment