Phil Wang 1307b3115d add exponential moving average of model, also allow passing in custom noise schedule

2020-09-08 14:58:09 -07:00

..

__init__.py

offer training class to easily train model off an image directory

2020-09-06 14:22:44 -07:00

denoising_diffusion_pytorch.py

add exponential moving average of model, also allow passing in custom noise schedule

2020-09-08 14:58:09 -07:00