search

SciML / NeuralOperators.jl

DeepONets, (Fourier) Neural Operators, Physics-Informed Neural Operators, and more in Julia
https://docs.sciml.ai/NeuralOperators/stable/
MIT License
262 stars 41 forks source link

Consider using fft to rfft #151

Open dynamic-queries opened 10 months ago

dynamic-queries commented 10 months ago

I ran into this issue a few months ago. So apologies for the late report.

For the FNO, rfft is used. From https://github.com/FluxML/Zygote.jl/issues/1406 , see that Zygote's gradient is wrong. A quick fix is to change this to fft for now.

ChrisRackauckas commented 10 months ago

That seems like a fine temporary workaround. I'd accept it.