Merge pull request #10 from lbin/pytorch_1.7 Pytorch 1.7
support pytorch 1.0
"support version pytorch 0.4.0"