/intro-to-pytorch

Deep Learning Scripts for Everyone

Primary LanguageJupyter Notebook

Introduction to PyTorch

mnist.py

Classifies between handwritten digits from the famous MNIST dataset. Uses a single linear layer to predict, and uses cross-entropy to measure the loss.

This project is going to grow to include various new exciting models that span image classification problems. So stay tuned!