autopush

Author: SCiarella

configs/snellius64/fno_base.yaml

@@ -1,7 +1,7 @@
 docreatedata: true
 docomp: true
 ntrajectory: 8
-T: "Float64"
+T: "Float32"
 dataproj: true
 projtest: false
 params:
@@ -34,7 +34,7 @@ closure:
   rng: "Xoshiro(seeds.θ_start)"
 priori:
   dotrain: true
-  nepoch: 15000
+  nepoch: 150
   batchsize: 64
   opt: "Adam(T(1.0e-3))"
   do_plot: false
@@ -43,7 +43,7 @@ priori:
 posteriori:
   dotrain: true
   projectorders: "(ProjectOrder.Last, )"
-  nepoch: 3000
+  nepoch: 30
   opt: "Adam(T(1.0e-4))"
   nunroll: 5
   nunroll_valid: 10

extra_model_workflow.jl

@@ -310,6 +310,7 @@ let
     dotrain = conf["priori"]["dotrain"]
     nepoch = conf["priori"]["nepoch"]
     dotrain && trainprior(;
+        T,
         params,
         priorseed = seeds.prior,
         dns_seeds_train,
@@ -418,6 +419,7 @@ let
     dotrain = conf["posteriori"]["dotrain"]
     nepoch = conf["posteriori"]["nepoch"]
     dotrain && trainpost(;
+        T,
         params,
         projectorders,
         outdir,