I’m currently using a PyTorch version compiled from master.
Let me check the code with 0.4.0.
EDIT: It’s also working on 0.4.0. Which PyTorch version do you have?
Could you update to the current stable release? You will find the install instructions on the website 2.