Update about loss function. - Githubissues

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 about loss function. #79

Closed chairc closed 4 months ago

chairc commented 4 months ago

Add: Add HuberLoss() and SmoothL1Loss() loss functions. Update: Update choices method for parser. Update: Update loss function custom method during training.