PyTorch is definitely easier to use than Tensorflow. To install PyTorch: ran the following instructions from pytorch website:

Got as far as I could on tensorflow on cpu. Ran the following to get tensorflow in the notebook:

Updating to python 3.6 and spring 2017.

Starting on 2017 spring version

Implement spatial batchnorm

More progress on ConvNet

Fix the 'adam' implementation. Had to check the paper for the correct algorithm.

Backing up my work on assignment 2, stuck on "Overfit small data"