Train consistency models from scratch and sample with 1 step on CIFAR-10. Code is written in PyTorch.
### Train
$ python train.py --output_folder checkpoint/cifar10-cm
### Sampling
$ python sample.py --model_path checkpoint/cifar10-cm/last.pth-
Notifications
You must be signed in to change notification settings - Fork 0
SunnerLi/CM-CIFAR10
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Code of ICML 2023 paper. Train with less than 150 lines of code
Resources
Whoa there!
You have triggered an abuse detection mechanism.
Please wait a few minutes before you try again;
in some cases this may take up to an hour.
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published