CM-CIFAR10

Train consistency models from scratch and sample with 1 step on CIFAR-10. Code is written in PyTorch.

Usage

### Train
$ python train.py --output_folder checkpoint/cifar10-cm
### Sampling
$ python sample.py --model_path checkpoint/cifar10-cm/last.pth

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
readme.md		readme.md
sample.py		sample.py
train.py		train.py
unet.py		unet.py