This is a full training cd /Users/leo/venv/pytorch_run/hw1_part2/train_all_cnn.py recitations: https://github.com/cmudeeplearning11785/Fall2018-tutorials/blob/master/recitation-3/11785-Fall18-Recitation-3.ipynb