Issue #2, make it compatible with later version#10

tiansiyuan wants to merge 2 commits intoswarmapytorch:masterswarmapytorch/book_DeepLearning_in_PyTorch_Source:masterfrom

tiansiyuan:mastertiansiyuan/book_DeepLearning_in_PyTorch_Source:masterCopy head branch name to clipboard