Question about parameter weights

zoubohao / DenoisingDiffusionProbabilityModel-ddpm-

This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.

MIT License

1.48k stars 156 forks source link

Question about parameter weights #20

Open Celia0u0 opened 1 year ago

Celia0u0 commented 1 year ago

May I ask, do the U-nets used in different denoising steps in the diffusion model share weights?

zoubohao commented 1 year ago

Yes, those weights are shared.