DL-Art-School/codes/models/diffusion
James Betker 6c6e82406e Pass a corruption factor through the dataset into the upsampling network
The intuition is this will help guide the network to make better informed decisions
about how it performs upsampling based on how it perceives the underlying content.

(I'm giving up on letting networks detect their own quality - I'm not convinced it is
actually feasible)
2021-06-07 09:13:54 -06:00
..
__init__.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
fp16_util.py Support diffusion unet 2021-06-06 13:57:22 -06:00
gaussian_diffusion.py A few fixes: 2021-06-05 13:40:32 -06:00
losses.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
nn.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
resample.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
respace.py GD mods & fixes 2021-06-04 17:13:16 -06:00
rrdb_diffusion.py Only head norm removed 2021-06-05 23:29:11 -06:00
unet_diffusion.py Pass a corruption factor through the dataset into the upsampling network 2021-06-07 09:13:54 -06:00