Issue in loss function

[SeongJinAhn]

It's very interesting to see your work.

Is your loss only consider a simplified objective function?

DPM-SNC/DPM-GSP-semi-supervised/method_series/gaussian_ddpm_losses.py

Line 91 in 46ea3d0

torch.sum(((pred_y[train_mask]-noise[train_mask])**2), dim = -1)

I think that a variable is an epsilon, not a \hat{\mu} because of :

DPM-SNC/DPM-GSP-semi-supervised/method_series/gaussian_ddpm_losses.py

Line 91 in 46ea3d0

torch.sum(((pred_y[train_mask]-noise[train_mask])**2), dim = -1)