How to implement the training objective for predicted variance?

[LiJunnan1992]

Thanks for the great model!

I notice that the unet predicts a learned variance. Any guidance on how to implement the training objective for the variance?

Thanks, Junnan