Aboat Loss Functions

[Jason2Lau]

Thank you for your awesome work. I have some questions about the loss function.
The diffusion loss is defined as follows in the original paper:

However, in your code, the diffusion loss uses a form similar to the reconstruction loss:

Which one is correct?

[Ephemeral182]

Hi, you may replace the PSNR loss with an L2 loss, as they are based on the same principle.