Phil Wang 2b742dd2cc move accelerator backward outside of autocast context, also calculate total loss correctly across gradient accumulated steps

2022-07-11 21:02:06 -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

bring in ddim sampling

2022-07-09 16:10:23 -07:00

denoising_diffusion_pytorch.py

move accelerator backward outside of autocast context, also calculate total loss correctly across gradient accumulated steps

2022-07-11 21:02:06 -07:00

elucidated_diffusion.py

reorganize elucidating code

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

learned_gaussian_diffusion.py

bring in ddim sampling

2022-07-09 16:10:23 -07:00

weighted_objective_gaussian_diffusion.py

bring in ddim sampling

2022-07-09 16:10:23 -07:00