This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
MIT License
1.48k
stars
156
forks
source link
I really love the idea you just put attn block in resblock and use a parameter to control whether to use it. #41
I was really confused about when and how to place attn blocks in other people code or tutorial.
and I found you just put it in ResBlock, and it's much easier for us to manage and it.
I was really confused about when and how to place attn blocks in other people code or tutorial. and I found you just put it in ResBlock, and it's much easier for us to manage and it.