Creating a diffusion model from scratch in PyTorch and training it for denoising Path Tracing renderings.
- Implement the diffusion model in PyTorch.
- Implement the training loop.
- Overfit on the Cornell Box dataset generated using mitusba.
- Improve speed using https://github.com/hutaiHang/Faster-Diffusion
- https://arxiv.org/html/2307.03992v3/#S5 Diffusion for General Denoising