Add error handling to training process

[tk-king]

Check the training code in the backend and add error handling everywhere there might go something wrong, so that the user will get detailed feedback when training a model