openai / improved-diffusion

Release for Improved Denoising Diffusion Probabilistic Models
MIT License
3.27k stars 488 forks source link

Which dataset is Upsampling 256x256 model trained on? #75

Open caihaunqai opened 1 year ago

caihaunqai commented 1 year ago

Thank you for your excellent job. I wonder which dataset is "Upsampling 256x256 model (280M parameters, trained for 500K iterations)" is trained on?

wgsthink commented 1 year ago

@caihaunqai Have you successfully generated images based on the Upsampling 256x256 model? I found that in this model, the out channel number is 6, why? I think that should be 3, like the RGB figure, hope to receive your response, thank you.

wgsthink commented 1 year ago

I get it, the first 3 channels are the mean of X{t-1}, the last 3 channels are the variance of X{t-1}