search

Easquel / WaveDM

Implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.
https://ieeexplore.ieee.org/abstract/document/10420512
19 stars 1 forks source link

training the hfrm #6

Open Aaronwangz opened 2 weeks ago

Aaronwangz commented 2 weeks ago

Thank you very much for sharing, I don't know if I didn't study your code carefully enough. I found that you didn't seem to use the wavelet transform before training the hfrm module, it seems to be a little different from the original text, can you give me an answer?

Easquel commented 5 days ago

Thanks for your attention. Later experiments show that this option can achieve a little bit faster training than the other one while they both obtain similar performance.