RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [128]] is at version 2; expected version 1 instead.

wangguanzhi / LADN

This is the implementation for Local Adversarial Disentangling Network for Facial Makeup and De-Makeup

https://georgegu1997.github.io/LADN-project-page/

181 stars 26 forks source link

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [128]] is at version 2; expected version 1 instead. #16

Open tianxiaoer opened 1 year ago

tianxiaoer commented 1 year ago

Hello, I encountered the following problem while reproducing the code. Can you help me solve it.

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [128]] is at version 2; expected version 1 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).