training problem

[ZH3nGsQ]

hello! when I am training with one gpu. I encountered with a problem:

and I set the script like this

I don't know which leaf Variable that requires grad is being used in an in-place operation.

[gasvn]

Can you show me how to reproduce the error?

[ZH3nGsQ]

I just run the code block and this error occurred.