This is the code for "PyTorch in 5 Minutes" by Siraj Raval on Youtube
This is the code for this video on Youtube by Siraj Raval. It's meant to be a demonstration of PyTorch's key features (dynamic computation graphs and imperative programming). This script builds a simple 2 layer neural net that learns the mapping between randomly generated input and output data.
- pytorch (Install it using the instructions here)
Run python demo.py in terminal to see the results
Credits go to jcohnson. I've merely created a wrapper to get people started.