SciML / OperatorLearning.jl

No need to train, he's a smooth operator
https://operatorlearning.sciml.ai/dev
MIT License
43 stars 8 forks source link

Flux.trainable - Handling trivial return case #39

Open dynamic-queries opened 2 years ago

dynamic-queries commented 2 years ago

Issue :
From the example for the FNO, collecting parameters throws an error as such. image

Potential cause: In https://github.com/SciML/OperatorLearning.jl/blob/master/src/FourierLayer.jl#L55-L56 and subsequently https://github.com/FluxML/Flux.jl/blob/master/src/utils.jl#L500-L502, there is trivial Bool return that FourierLayer.jl does not handle, therefore throwing an error.