Ryan Burgert 7c0cd05c27 Trainer.load can use GPU's other than cuda:0

2022-09-04 22:11:35 -04:00

..

__init__.py

some basic scaffold for elucidating diffusion and derived values

2022-06-28 13:11:56 -07:00

continuous_time_gaussian_diffusion.py

add the new self conditioning technique from hintons group from bit diffusion paper

2022-08-10 10:52:07 -07:00

denoising_diffusion_pytorch.py

Trainer.load can use GPU's other than cuda:0

2022-09-04 22:11:35 -04:00

elucidated_diffusion.py

add self conditioning for elucidated ddpm

2022-08-10 13:15:45 -07:00

learned_gaussian_diffusion.py

add the new self conditioning technique from hintons group from bit diffusion paper

2022-08-10 10:52:07 -07:00

weighted_objective_gaussian_diffusion.py

add the new self conditioning technique from hintons group from bit diffusion paper

2022-08-10 10:52:07 -07:00