SunnerLi / CM-CIFAR10 Public

Notifications You must be signed in to change notification settings
Fork 0
Star 2

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

2 stars 0 forks Branches Tags Activity

Notifications

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

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

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%