RunRuntimeError because of an inplace operation

[ptrblck]

Good to hear it’s working now. I was referring to this issue which raises the same error.