From d22b5236a89ac4261893e1021df777dc2b64fab4 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Mon, 19 Aug 2024 21:19:20 -0700 Subject: [PATCH] fix: accidental public API change (#26) --- src/fno.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/fno.jl b/src/fno.jl index e5f7cfd..511d2e0 100644 --- a/src/fno.jl +++ b/src/fno.jl @@ -27,7 +27,7 @@ kernels, and two `Dense` layers to project data back to the scalar field of inte ## Example ```jldoctest -julia> fno = FourierNeuralOperator(; σ=gelu, chs=(2, 64, 64, 128, 1), modes=(16,)); +julia> fno = FourierNeuralOperator(gelu; chs=(2, 64, 64, 128, 1), modes=(16,)); julia> ps, st = Lux.setup(Xoshiro(), fno); @@ -44,8 +44,8 @@ julia> size(first(fno(u, ps, st))) project end -function FourierNeuralOperator(; - σ=gelu, chs::Dims{C}=(2, 64, 64, 64, 64, 64, 128, 1), modes::Dims{M}=(16,), +function FourierNeuralOperator( + σ=gelu; chs::Dims{C}=(2, 64, 64, 64, 64, 64, 128, 1), modes::Dims{M}=(16,), permuted::Val{perm}=Val(false), kwargs...) where {C, M, perm} @argcheck length(chs) ≥ 5