learning gluon with mnist dataset - View it on GitHub