Fixing some logging and the path to the training data.