Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit d22b523

Browse files
authored
fix: accidental public API change (#26)
1 parent e248a56 commit d22b523

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/fno.jl

+3-3
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ kernels, and two `Dense` layers to project data back to the scalar field of inte
2727
## Example
2828
2929
```jldoctest
30-
julia> fno = FourierNeuralOperator(; σ=gelu, chs=(2, 64, 64, 128, 1), modes=(16,));
30+
julia> fno = FourierNeuralOperator(gelu; chs=(2, 64, 64, 128, 1), modes=(16,));
3131
3232
julia> ps, st = Lux.setup(Xoshiro(), fno);
3333
@@ -44,8 +44,8 @@ julia> size(first(fno(u, ps, st)))
4444
project
4545
end
4646

47-
function FourierNeuralOperator(;
48-
σ=gelu, chs::Dims{C}=(2, 64, 64, 64, 64, 64, 128, 1), modes::Dims{M}=(16,),
47+
function FourierNeuralOperator(
48+
σ=gelu; chs::Dims{C}=(2, 64, 64, 64, 64, 64, 128, 1), modes::Dims{M}=(16,),
4949
permuted::Val{perm}=Val(false), kwargs...) where {C, M, perm}
5050
@argcheck length(chs) 5
5151

0 commit comments

Comments
 (0)