This project was made in collaboration with @PhilipTamb for the course "Cognitive Computing & Artificial Intelligence" @ University of Catania.
The file contains a Pytorch code for training the diffusion model described by this paper. We're also using the code by @lucidrain for the implementation of Unet, Gaussian Diffusion and the trainer class.
ATTENTION! This code only works with Google Colab, if you want to try it on your own GPU you must do some adjustments.
For more details, you can read the documentation which is the pdf file of the repo (don't worry, this time is in english :P )
vinz97/Denoising-diffusion-replication
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|