This is the error I'm getting if I'm trying to run `test.py` with PyTorch **1.1** It's just a heads-up - this doesn't seem to be a problem on your end. Downgrading to PyTorch **1.0.1** solved the issue for me.