Select Git revision
docs
Open with
- Visual Studio Code
- IntelliJ IDEA
Download source code
Download directory
Your workspaces
A workspace is a virtual sandbox environment for your code in GitLab.
No agents available to create workspaces. Please consult Workspaces documentation for troubleshooting.
Gonzalo Martin Garcia authored
Completed Class Conditional Diffusion Model. This version is intended to be trained on a class labeled dataset. It makes use of classifier-free guided diffusion to boost sample quality of the model when generating images from each class. The conditioning mechanism in the UNet (implemented together with Roy) simply adds an embedding of the class to the time embedding before passing them through the learnable reshaping layers for each block. Successfully trained on 3 class dataset, dogs, cats and wildlife.
Code owners
Assign users and groups as approvers for specific file changes. Learn more.
Name | Last commit | Last update |
---|---|---|
.. | ||
dataloader | ||
evaluation | ||
models | ||
trainer | ||
.gitignore | ||
README.md | ||
__init__.py | ||
experiment_creator.ipynb | ||
main.py |