Scatterline for epost(t)

Author: SCiarella

configs/local/fno_1.yaml

@@ -0,0 +1,49 @@
+docreatedata: true
+docomp: false
+ntrajectory: 8
+T: "Float32"
+params:
+  D: 2
+  lims: [0.0, 1.0]
+  Re: 6000.0
+  tburn: 0.5
+  tsim: 2.0
+  savefreq: 100
+  ndns: 1024
+  nles: [32]
+  filters: ["FaceAverage()"]
+  icfunc: "(setup, psolver, rng) -> random_field(setup, T(0); kp=20, psolver, rng)"
+  method: "RKMethods.Wray3(; T)"
+  bodyforce: "(dim, x, y, t) -> (dim == 1) * 5 * sinpi(8 * y)"
+  issteadybodyforce: true
+  processors: "(; log = timelogger(; nupdate=100))"
+  Δt: 0.0001
+seeds:
+  dns: 12345
+  θ_start: 234
+  prior: 345
+  post: 456
+closure:
+  name: "fno_1"
+  type: fno
+  channels: [2, 64, 64, 64, 2]
+  modes: [16, 16]
+  activation: "Lux.gelu"
+  rng: "Xoshiro(seeds.θ_start)"
+priori:
+  dotrain: true
+  nepoch: 5000
+  batchsize: 64
+  opt: "OptimiserChain(Adam(T(1.0e-3)), ClipGrad(0.1))"
+  do_plot: false
+  plot_train: false
+posteriori:
+  dotrain: true
+  projectorders: "(ProjectOrder.Last, )"
+  nepoch: 300
+  opt: "OptimiserChain(Adam(T(1.0e-3)), ClipGrad(0.1))"
+  nunroll: 5
+  nunroll_valid: 5
+  dt: T(1e-4)
+  do_plot: false
+  plot_train: false

src/plots.jl

@@ -623,65 +623,70 @@ function plot_epost_vs_t(error_file, closure_name, nles, ax, color, PLOT_STYLES)
     x = error_data.nts  # use time from error data
 
     # Prior
-    lines!(
+    scatterlines!(
         ax,
         x,
         vec(error_data.model_prior);
         label = "$closure_name prior (n = $nles)",
         color = color,
         linestyle = PLOT_STYLES[:prior].linestyle,
         linewidth = PLOT_STYLES[:prior].linewidth,
+        marker = :circle,
     )
 
     # Post
-    lines!(
+    scatterlines!(
         ax,
         x,
         vec(error_data.model_post);
         label = "$closure_name post (n = $nles)",
         color = color,
         linestyle = PLOT_STYLES[:post].linestyle,
         linewidth = PLOT_STYLES[:post].linewidth,
+        marker = :circle,
     )
 
     # Smagorinsky (optional)
     if haskey(error_data, Symbol("smag"))
-        lines!(
+        scatterlines!(
             ax,
             x,
             vec(error_data.smag);
             label = "$closure_name smag (n = $nles)",
             color = PLOT_STYLES[:smag].color,
             linestyle = PLOT_STYLES[:smag].linestyle,
             linewidth = PLOT_STYLES[:smag].linewidth,
+            marker = :circle,
         )
     end
 
     if closure_name == "INS_ref"
         label = "No model"
         if _missing_label(ax, label)  # add No closure only once
-            lines!(
+            scatterlines!(
                 ax,
                 x,
                 vec(error_data.nomodel);
                 label = label,
                 linestyle = PLOT_STYLES[:no_closure].linestyle,
                 linewidth = PLOT_STYLES[:no_closure].linewidth,
                 color = PLOT_STYLES[:no_closure].color,
+                marker = :circle,
             )
         end
     end
 
     label = "No model (projected dyn)"
     if _missing_label(ax, label)  # add No closure only once
-        lines!(
+        scatterlines!(
             ax,
             x,
             vec(error_data.nomodel);
             label = label,
             linestyle = PLOT_STYLES[:no_closure_proj].linestyle,
             linewidth = PLOT_STYLES[:no_closure_proj].linewidth,
             color = PLOT_STYLES[:no_closure_proj].color,
+            marker = :circle,
         )
     end