version 1.0
train script: train_script.py sample script: sample_script.py
still remains trouble
precision default to fp16, SGD optimization, xformers, image_size & channel dim set to 256.(OOM if set to 512...) Why does GPU memory still exceed 20G?
-
Notifications
You must be signed in to change notification settings - Fork 0
00drdelius/CustomDDPM
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
This is a custom DDPM(Denoising Diffusion Probablistic Model).
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published