Phil Wang cee30ce304 optimizer should be saved and loaded

2022-07-08 17:07:32 -07: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

make sure all versions of torch supported

2022-06-23 12:28:09 -07:00

denoising_diffusion_pytorch.py

optimizer should be saved and loaded

2022-07-08 17:07:32 -07:00

elucidated_diffusion.py

reorganize elucidating code

2022-06-29 08:49:38 -07:00

learned_gaussian_diffusion.py

add convert_image_to keyword argument, for forcing images being loaded to be converted to some format, greyscale, rgb, rgba, whatever

2022-07-08 09:22:59 -07:00

weighted_objective_gaussian_diffusion.py

add gaussian diffusion where model predicts both noise and x_start, with a learned weighting between the two (experimental)

2022-05-13 13:56:32 -07:00