yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
https://arxiv.org/abs/2011.13456
Apache License 2.0
1.58k stars 295 forks source link

How to do temperature rescaling #13

Open Lifedecoder opened 2 years ago

Lifedecoder commented 2 years ago

Hi! Thanks for this marvellous work. I have questions about how to do temperature scaling, like Fig. 6 in the paper does. I use pretrained ODE model to generate 256x256 Celeba images, but the results are too smooth and blurry. I wonder how I can get more realistic pics like Fig. 6 does? Specifically how to "reduce the norm of embedding" to do temperature scaling?

Many thanks.