Ting-Chun Wang add pytorch 0.4.1 support

Latest commit bdb7ec6

Aug 23, 2018