Correct training error (Mistral)

Related categories: Python / Artificial Intelligence