← all repositories

yang-song/score_sde_pytorch

A PyTorch implementation of score-based generative models using stochastic differential equations for high-quality image generation.

2.1k stars Jupyter Notebook Image · Video · AudioML Frameworks
score_sde_pytorch
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

This repository implements the score-based generative modeling framework through stochastic differential equations (SDEs), originally published at ICLR 2021. The method transforms data into noise via a continuous-time stochastic process, then reverses the SDE for sample generation using score matching. It achieves state-of-the-art FID of 2.20 on CIFAR-10 and generates high-fidelity 1024px Celeba-HQ images. The work supports various applications including class-conditional generation, inpainting, and colorization.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.