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

Hi, I have added many new features to your code #37

Closed chairc closed 5 months ago

chairc commented 11 months ago

Hi, I have added many new features to your code.

I want to express my gratitude for providing such a detailed explanation of DDPM. I have made numerous improvements and added functionalities to your code, including the following changes:

  1. Added support for custom image sizes.
  2. Included code segments for DDIM.
  3. Enhanced the generator and trainer methods for improved usability.
  4. Implemented cosine and warmup cosine learning rate schedules.
  5. Added support for distributed training, allowing for larger models to be trained across multiple GPUs.
  6. Introduced custom seeds for improved reproducibility.
  7. Enabled half-precision training to reduce GPU memory usage.
  8. Provided the ability to choose different optimizers.
  9. Added the option to select different activation functions for training.
  10. Implemented a training checkpoint recovery feature, allowing training to resume from an interruption point.
  11. Added support for server deployment, enabling access through an interface.
  12. Refactored the code to make it more modular.

I will continue to build upon your work by introducing more optimizations and new features. Thank you for your contributions to the community.

Project Repository: https://github.com/chairc/Industrial-Defect-Diffusion-Model

ZhaohuiQiao0517 commented 11 months ago

good work!