fix for working with PyTorch v1.0.1#16

solesensei wants to merge 1 commit intoadventuresinML:masteradventuresinML/adventures-in-ml-code:masterfrom

solesensei:mastersolesensei/adventures-in-ml-code:masterCopy head branch name to clipboard