Train a tiny diffusion model even with CPU. Implemented with Pytorch. Tiny U-net style model and mnist dataset using only label
0
and1
[TOC]
- python3.8
- pip install -r requirements.txt
Pretrained model trained with 20000 iterations. Result save in eval_out
More iterations should be better. Strongly recommend you try it.
bash eval.sh
bash train.sh
Feel free to tune the training arguments as you want. Have fun!
Welcome to pull request.