Revise and test FlowOverCircle

foldfelis · foldfelis · commit 0b7e030857b1 · 2022-03-07T12:50:16.000+08:00
diff --git a/example/FlowOverCircle/src/FlowOverCircle.jl b/example/FlowOverCircle/src/FlowOverCircle.jl
@@ -24,10 +24,10 @@ function train()
 
     m = Chain(
         Dense(1, 64),
-        OperatorKernel(64=>64, (24, 24), gelu),
-        OperatorKernel(64=>64, (24, 24), gelu),
-        OperatorKernel(64=>64, (24, 24), gelu),
-        OperatorKernel(64=>64, (24, 24), gelu),
+        OperatorKernel(64=>64, (24, 24), FourierTransform, gelu),
+        OperatorKernel(64=>64, (24, 24), FourierTransform, gelu),
+        OperatorKernel(64=>64, (24, 24), FourierTransform, gelu),
+        OperatorKernel(64=>64, (24, 24), FourierTransform, gelu),
         Dense(64, 1),
     ) |> device
 
