.

Author: SCiarella

benchmark.jl

@@ -19,7 +19,7 @@ ispath(compdir) || mkpath(compdir)
 # List configurations files
 using Glob
 #exclude_patterns = ["att", "cno", "cnn_ins", "_1", "nopr"]
-exclude_patterns = ["att", "cno", "int", "back", "rk4"]
+exclude_patterns = ["att", "cno", "int", "back", "rk4", "cnn_1"]
 @warn "Excluding configurations with patterns: $(exclude_patterns)"
 all_confs = glob("*.yaml", confdir)
 list_confs = filter(conf -> all(!occursin(pat, conf) for pat in exclude_patterns), all_confs)
@@ -101,16 +101,16 @@ colors_list = [
 
 # Loop over plot types and configurations
 plot_labels = Dict(
-    :prior_hist => (
-        title  = "A-priori training history for different configurations",
-        xlabel = "Iteration",
-        ylabel = "A-priori error",
-    ),
-    :posteriori_hist => (
-        title  = "A-posteriori training history for different configurations",
-        xlabel = "Iteration",
-        ylabel = "DCF",
-    ),
+    #:prior_hist => (
+    #    title  = "A-priori training history for different configurations",
+    #    xlabel = "Iteration",
+    #    ylabel = "A-priori error",
+    #),
+    #:posteriori_hist => (
+    #    title  = "A-posteriori training history for different configurations",
+    #    xlabel = "Iteration",
+    #    ylabel = "DCF",
+    #),
     :divergence => (
         title  = "Divergence for different configurations",
         xlabel = "t",

configs/snellius/cnn_noproj.yaml

@@ -0,0 +1,53 @@
+docreatedata: true
+docomp: true
+ntrajectory: 8
+T: "Float32"
+dataproj: false
+params:
+  D: 2
+  lims: [0.0, 1.0]
+  Re: 6000.0
+  tburn: 0.5
+  tsim: 5.0
+  savefreq: 50
+  ndns: 4096
+  nles: [64]
+  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.00005
+seeds:
+  dns: 123456
+  θ_start: 234
+  prior: 345
+  post: 456
+closure:
+  name: "NoProjection"
+  type: cnn
+  radii: [2, 2, 2, 2, 2]
+  channels: [24, 24, 24, 24, 2]
+  activations: ["tanh", "tanh", "tanh", "tanh", "identity"]
+  use_bias: [true, true, true, true, false]
+  rng: "Xoshiro(seeds.θ_start)"
+priori:
+  dotrain: true
+  nepoch: 50000
+  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: 3000
+  opt: "Adam(T(1.0e-4))"
+  nunroll: 5
+  nunroll_valid: 10
+  dt: 0.0001
+  do_plot: false
+  plot_train: false
+  nsamples: 5
+  sciml_solver: "Tsit5()"

multisub.sh

@@ -5,3 +5,6 @@ sbatch -J CNN job_a100.sh configs/snellius/cnn_1.yaml
 sbatch -J backsol job_a100.sh configs/snellius/cnn_backsol.yaml
 sbatch -J interp job_a100.sh configs/snellius/cnn_interp.yaml
 sbatch -J rk4 job_a100.sh configs/snellius/cnn_rk4.yaml
+
+sbatch -J CNN_proj job_a100.sh configs/snellius/cnn_proj.yaml
+sbatch -J CNN_nopr job_a100.sh configs/snellius/cnn_noproj.yaml

src/plots.jl

@@ -254,7 +254,7 @@ function plot_energy_evolution(
         end
     end
 
-    if closure_name == "cnn_proj"
+    if closure_name == "Project"
         label = "No closure (projected dyn)"
         if _missing_label(ax, label) && haskey(energyhistory, Symbol("nomodel"))
             lines!(
@@ -361,7 +361,7 @@ function plot_energy_evolution_hist(
         end
     end
 
-    if closure_name == "cnn_1"
+    if closure_name == "Project"
         label = "No closure (projected dyn)"
         if _missing_label(ax, label) && haskey(energyhistory, Symbol("nomodel"))
             _plot_histogram(
@@ -868,7 +868,7 @@ function plot_epost_vs_t(error_file, closure_name, nles, ax, color, PLOT_STYLES)
         end
     end
 
-    if closure_name == "cnn_noproj"
+    if closure_name == "NoProjection"
         label = "No model (projected dyn)"
         if _missing_label(ax, label)  # add No closure only once
             scatterlines!(