Instructions for setting up:
- Clone the repo or download the zip.
- Go to https://colab.research.google.com/ and open the ipynb notebook
- Go to runtime environment and set it to use the GPU.
- Install pytorch on colab if you get an error: !pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl
- You should be good to run the code!