autopush

Author: SCiarella

cnn_model_workflow.jl

@@ -177,7 +177,6 @@ for i = 1:ntrajectory
 	end
 end
 @info "Data generated"
-@assert 1==2
 
 # Computational time
 docomp = conf["docomp"]

configs/bck_snell/att_1.yaml

@@ -25,7 +25,7 @@ seeds:
   prior: 345
   post: 456
 closure:
-  name: "attention_reg"
+  name: "attention"
   type: attentioncnn
   radii: [2, 2, 2, 2, 2]
   channels: [24, 24, 24, 24, 2]
@@ -42,7 +42,7 @@ closure:
   rng: "Xoshiro(seeds.θ_start)"
 priori:
   dotrain: true
-  nepoch: 5000
+  nepoch: 50000
   batchsize: 64
   opt: "OptimiserChain(Adam(T(1.0e-3)), ClipGrad(0.1))"
   do_plot: false

job_a100.sh

@@ -8,7 +8,7 @@
 #SBATCH --time=05:00:00
 #SBATCH --mail-type=BEGIN,END
 # #SBATCH [email protected]
-#SBATCH --array=1-8
+#SBATCH --array=1-1
 
 module load 2023
 module load juliaup/1.14.5-GCCcore-12.3.0