yunik1004 / SAiD

SAiD: Blendshape-based Audio-Driven Speech Animation with Diffusion
https://yunik1004.github.io/SAiD/
Apache License 2.0
74 stars 14 forks source link

Train Error #8

Open xxxukk opened 4 months ago

xxxukk commented 4 months ago

Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [8, 32, 120]], which is output 0 of TanhBackward0, is at version 1; expected version 0 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

xxxukk commented 3 months ago

遇到这个问题将train代码中的+=与*=修改为普通的加法减法