learning pytorch with mnist dataset - View it on GitHub