The last two things to setup before you start training is to compute the accuracy from the predictions, and provide a way to push your model to the Hub.