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.
predict.py
end_token
==
is