Fix CER/WER always equal to 1.0
Closes #24 (closed)
As in the predict.py
code (see https://gitlab.com/teklia/atr/dan/-/blob/main/dan/predict.py#L153), to check if the end_token
has been predicted, we should use ==
instead of is
since it doesn't work correctly on tensors.
Edited by Mélodie Boillet
Merge request reports
Activity
changed milestone to %ML Prod - December 2022 n°2
added P2 label
Please register or sign in to reply