search

dome272 / Diffusion-Models-pytorch

Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)
Apache License 2.0
1.11k stars 256 forks source link

Fix unet upsampling dimensions #18

Open davegrays opened 1 year ago

davegrays commented 1 year ago

the channel dimensions for the unet were doing some weird things (e.g. decreasing by 4x then back to 2x) I believe this is more consistent with the original unet implementation