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

Update: Refactor train.py format. #66

Closed chairc closed 6 months ago

chairc commented 6 months ago

format.Update: Refactor train.py format.