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

[AaronYALai (Aaron Lai)]

Nice explanation!!
I encountered the same problem and solved it by the explanation!
Thanks