OperatorLearning.jl icon indicating copy to clipboard operation
OperatorLearning.jl copied to clipboard

Flux.trainable - Handling trivial return case

Open dynamic-queries opened this issue 3 years ago • 0 comments

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.

dynamic-queries avatar Jun 14 '22 08:06 dynamic-queries