Károly Harsányi

There is a problem with the implementation of the ```update_train_state``` function in chapters 3-5. Specifically, when the loss is decreasing ```train_state['early_stopping_best_val']``` is not updated (except in the first epoch), so...