Wavelet loss function?

[nitinh12]

Is wavelet loss function supported for flux? @kohya-ss

[rockerBOO]

https://arxiv.org/abs/2306.00306 https://arxiv.org/abs/2402.19215 https://arxiv.org/abs/2211.16152 https://arxiv.org/abs/2407.12538

Looking into this but some related papers.

[nitinh12]

https://arxiv.org/abs/2306.00306 https://arxiv.org/abs/2402.19215 https://arxiv.org/abs/2211.16152 https://arxiv.org/abs/2407.12538

Looking into this but some related papers.

Thanks for sharing these. It would be great if this gets added for flux training. AI toolkit recently implemented this.

[nitinh12]

@rockerBOO Any update on your PR?

[rockerBOO]

@nitinh12 I am using it every day, multiple times a day. It is working as expected but I have been testing some other things. Also waiting to hear from a few people who are trying it.