Okay. I’m running on the master version. I think pytorch has gotten better support for in-place operations since the 0.3 release. You can try building from source (instructions here 10) to use the latest features.