getting rid of the try except loop to have the train.main_loop() inform the user
of the underlying error (if there is a bug or an error there)