RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [256]], which is output 0 of AsStrid edBackward0, is at version 4; expected version 3 instead

autograd
3

Yeah that’s the problem. Thanks a lot!