Fashion-MNIST classifier training code