DL-Art-School/codes/models/diffusion
James Betker 80d4404367 A few fixes:
- Output better prediction of xstart from eps
- Support LossAwareSampler
- Support AdamW
2021-06-05 13:40:32 -06:00
..
__init__.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
gaussian_diffusion.py A few fixes: 2021-06-05 13:40: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
respace.py GD mods & fixes 2021-06-04 17:13:16 -06:00
rrdb_diffusion.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00