This is the training code and some public data