Skip to content

Fix CER/WER always equal to 1.0

Mélodie Boillet requested to merge fix-training-evaluation into main

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