add test for train

Author: foldfelis

src/fourier.jl

@@ -4,6 +4,7 @@ using Tullio
 
 export
     SpectralConv1d,
+    FourierOperator,
     FNO
 
 struct SpectralConv1d{T,S}
@@ -74,5 +75,3 @@ function FNO()
         flatten
     )
 end
-
-# loss(m::SpectralConv1d, x, x̂) = sum(abs2, x̂ .- m(x)) / len

test/fourier.jl

@@ -15,5 +15,11 @@ end
 
 @testset "FNO" begin
     𝐱, 𝐲 = get_data()
+    𝐱, 𝐲 = Float32.(𝐱), Float32.(𝐲)
     @test size(FNO()(𝐱)) == size(𝐲)
+
+    # m = FNO()
+    # loss(𝐱, 𝐲) = sum(abs2, 𝐲 .- m(𝐱)) / size(𝐱)[end]
+    # data = [(𝐱[:, :, 1:5], 𝐲[:, 1:5])]
+    # Flux.train!(loss, params(m), data, Flux.ADAM())
 end