DL-Art-School/codes/models/diffusion
James Betker 6084915af8 Support gaussian diffusion models
Adds support for GD models, courtesy of some maths from openai.

Also:
- Fixes requirement for eval{} even when it isn't being used
- Adds support for denormalizing an imagenet norm
2021-06-02 21:47:32 -06:00
..
__init__.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
gaussian_diffusion.py Support gaussian diffusion models 2021-06-02 21:47: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
rrdb_diffusion.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00