Bug in target noise
rahulvigneswaran opened this issue · 1 comments
rahulvigneswaran commented
You are doing MSE(noise, predicted_noise)
. Shouldn't it be MSE(coef2*noise, predicted_noise)
?
ddpm-torch/ddpm_torch/diffusion.py
Line 235 in 1bf0e5d