add extra confs

Author: SCiarella

benchmark.jl

@@ -18,7 +18,8 @@ ispath(compdir) || mkpath(compdir)
 
 # List configurations files
 using Glob
-exclude_patterns = ["att", "cno", "cnn_ins", "_1"]
+#exclude_patterns = ["att", "cno", "cnn_ins", "_1", "nopr"]
+exclude_patterns = ["att", "cno", "int", "back", "rk4"]
 @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)

configs/snellius/cnn_gauss.yaml

@@ -0,0 +1,55 @@
+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: "cnn_gauss"
+  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:
+  reuse: "cnn_noproj"
+  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: 1500
+  opt: "OptimiserChain(Adam(T(1.0e-4)), ClipGrad(0.01))"
+  nunroll: 5
+  nunroll_valid: 10
+  nsamples: 1
+  dt: 0.0001
+  do_plot: false
+  plot_train: false
+  sensealg: "GaussAdjoint()"
+  sciml_solver: "Tsit5()"

configs/snellius/cnn_ins.yaml

@@ -25,7 +25,7 @@ seeds:
   prior: 345
   post: 456
 closure:
-  name: "cnn_noproj"
+  name: "NoProjection"
   type: cnn
   radii: [2, 2, 2, 2, 2]
   channels: [24, 24, 24, 24, 2]
@@ -42,12 +42,12 @@ priori:
 posteriori:
   dotrain: true
   projectorders: "(ProjectOrder.Last, )"
-  nepoch: 1500
-  opt: "OptimiserChain(Adam(T(1.0e-3)), ClipGrad(0.1))"
+  nepoch: 3000
+  opt: "Adam(T(1.0e-4))"
   nunroll: 5
   nunroll_valid: 10
   dt: 0.0001
   do_plot: false
   plot_train: false
-  nsamples: 1
+  nsamples: 5
   sciml_solver: "Tsit5()"

configs/snellius/cnn_proj.yaml

@@ -0,0 +1,53 @@
+docreatedata: true
+docomp: true
+ntrajectory: 8
+T: "Float32"
+dataproj: true
+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: "Project"
+  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
+  nsamples: 5
+  dt: 0.0001
+  do_plot: false
+  plot_train: false
+  sciml_solver: "Tsit5()"

configs/snellius/cnn_rk4.yaml

@@ -25,7 +25,7 @@ seeds:
   prior: 345
   post: 456
 closure:
-  name: "cnn_noproj"
+  name: "cnn_rk4"
   type: cnn
   radii: [2, 2, 2, 2, 2]
   channels: [24, 24, 24, 24, 2]

multisub.sh

@@ -1,3 +1,7 @@
 #sbatch -J CNO job_a100_extra.sh configs/snellius/cno_1.yaml
 sbatch -J ATT job_a100_extra.sh configs/snellius/att_1.yaml
 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