Encounter the RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

[kevin-ssy (Shuyang Sun)]

Same problem. Have you found the solution?