I think the same is discussed here: Encounter the RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation