NVlabs / edm2

Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)
Other
479 stars 19 forks source link

Adding noise during sampling #1

Closed parlance-zz closed 5 months ago

parlance-zz commented 5 months ago

Hi guys,

Love the paper / model, just wanted to let you guys know you can (slightly) improve the fid even further by using the learned timestep uncertainty / logvar to add an appropriate amount of noise during the sampling / inference process at each timestep.