Predicted output of the model

[merryTong]

Hello, I read your paper and code. The paper describes the model Unet as being used to predict x0, but why is it predicting noise in the code?

[QiferG]

Hello, I read your paper and code. The paper describes the model Unet as being used to predict x0, but why is it predicting noise in the code?

Thanks to your question, U-Net can predict both noise and x0 directly. There is a conversion relationship between noise ε and x0, making them essentially equivalent. you can get more information here.

[merryTong]

I got it. Thanks.