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
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
dan/manager/training.py
+ 1
− 1
View file @ 5641f4c8
...@@ -1262,7 +1262,7 @@ class Manager(OCRManager): ...@@ -1262,7 +1262,7 @@ class Manager(OCRManager):
) )
predicted_tokens_len += 1 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): if torch.all(reached_end):
break break
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment