Name		Name	Last commit message	Last commit date
readme.md		readme.md
sample.py		sample.py
train.py		train.py
unet.py		unet.py

Repository files navigation

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

Ref

Official repository

About

Code of ICML 2023 paper. Train with less than 150 lines of code

2 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages