tqch/ddpm-torch

Bug in target noise

rahulvigneswaran opened this issue · 1 comments

You are doing MSE(noise, predicted_noise). Shouldn't it be MSE(coef2*noise, predicted_noise)?

target = noise

tqch commented

This is not a bug but is intended if you take a look at the pseudo code in the original paper and their official implementation link.