search

chairc / Integrated-Design-Diffusion-Model

IDDM (Industrial, landscape, animate, spectrogram...), support DDPM, DDIM, PLMS, webui and multi-GPU distributed training. Pytorch实现，生成模型，扩散模型，分布式训练
Apache License 2.0
152 stars 22 forks source link

Separate get dataset methods into data files; Add initialization operation. #67

Closed chairc closed 6 months ago

chairc commented 6 months ago

Update: Separate get dataset methods into data files. Add: Add initialization operation of utils. Add: Add initialization operation of modules. Add: Add initialization operation of networks. Add: Add initialization operation of diffusion samples. Add: Add initialization operation of config.