Implementation of Denoising Diffusion Probabilistic Models in PyTorch
Implementation of Denoising Diffusion Probabilistic Models in PyTorch
First please install tensorfn
pip install tensorfn
It is simple convenience library for machine learning experiments. Sorry for the inconvenience.
First prepare lmdb dataset:
python prepare_data.py --size [SIZES, e.g. 128,256] --out [LMDB NAME] [DATASET PATH]
Then run training looop!
python train.py --n_gpu [NUMBER OF GPUS FOR TRAINING] --conf config/diffusion.conf
Also, you can tried to use improved configurations from https://openreview.net/forum?id=-NEXDKk8gZ (Cosine beta schedule, more multi-headed self attention, affine modulation for time conditions, new training hyperparameters.)
python train.py --n_gpu [NUMBER OF GPUS FOR TRAINING] --conf config/improved.conf
Samples from FFHQ
