I've update the code to support both Python2 and Python3, PyTorch 1.0. If you want the old version code please checkout branch v0.3 ...