From 35346e42f5568aaa42e3a48cb27df757e9c5a813 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Syver=20D=C3=B8ving=20Agdestein?= <syverda@gmail.com>
Date: Sat, 19 Oct 2024 18:04:21 +0200
Subject: [PATCH] chore: update script

---
 lib/PaperDC/postanalysis.jl | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/PaperDC/postanalysis.jl b/lib/PaperDC/postanalysis.jl
index bdd7e92ef..181c81798 100644
--- a/lib/PaperDC/postanalysis.jl
+++ b/lib/PaperDC/postanalysis.jl
@@ -156,8 +156,8 @@ datafiles =
         "$datadir/seed=$(repr(seed))_filter=$(Φ)_nles=$(nles).jld2"
     end
 
-create_data = false
-create_data && for (iseed, seed) in enumerate(dns_seeds)
+createdata = false
+createdata && for (iseed, seed) in enumerate(dns_seeds)
     if isslurm && iseed != taskid
         # Each task does one initial condition
         @info "Skipping seed $seed on for $taskid"
@@ -176,7 +176,8 @@ create_data && for (iseed, seed) in enumerate(dns_seeds)
 end
 
 # Computational time
-let
+docomp = false
+docomp && let
     comptime, datasize = 0.0, 0.0
     for f in datafiles[1, 1, :]
         comptime += load(f, "comptime")
@@ -192,7 +193,7 @@ end
 # Build LES setup and assemble operators
 setups = map(
     nles -> Setup(;
-        x = ntuple(α -> LinRange(params.lims..., nles + 1), params.D),
+        x = ntuple(α -> range(params.lims..., nles + 1), params.D),
         params.Re,
         params.ArrayType,
         params.bodyforce,
@@ -373,12 +374,13 @@ for (iorder, projectorder) in enumerate(projectorders),
     rng = Xoshiro(seeds.post) # Same seed for all training setups
     setup = setups[ig]
     psolver = psolver_spectral(setup)
+    nsubstep = 5
     loss = create_loss_post(;
         setup,
         psolver,
         method = RKProject(params.method, projectorder),
         closure,
-        nupdate = 2, # Time steps per loss evaluation
+        nupdate = nsubstep, # Time steps per loss evaluation
     )
     data_train = namedtupleload.(datafiles[ig, ifil, itrain])
     data_valid = namedtupleload.(datafiles[ig, ifil, ivalid])
@@ -399,7 +401,7 @@ for (iorder, projectorder) in enumerate(projectorders),
                 psolver,
                 method = RKProject(params.method, projectorder),
                 closure_model = wrappedclosure(closure, setup),
-                nupdate = 2,
+                nupdate = nsubstep,
             );
             θ,
             figfile,