kill me

experiments/experiment.jl

@@ -36,7 +36,7 @@ Base.@kwdef struct Experiment
     finaliser::Function = Flux.softmax
     loss::Function = Flux.Losses.crossentropy
     train_parallel::Bool = false
-    reg_strength::Real = 0.5
+    reg_strength::Real = 0.1
 end
 
 "A container to hold the results of an experiment."

experiments/gmsc.sh

@@ -2,8 +2,8 @@
 
 #SBATCH --job-name="GMSC (ECCCo)"
 #SBATCH --time=3:00:00
-#SBATCH --ntasks=48
-#SBATCH --cpus-per-task=1
+#SBATCH --ntasks=12
+#SBATCH --cpus-per-task=4
 #SBATCH --partition=compute
 #SBATCH --mem-per-cpu=4GB
 #SBATCH --account=innovation
@@ -11,4 +11,4 @@
 
 module load 2023r1 openmpi
 
-srun julia --project=experiments experiments/run_experiments.jl -- data=gmsc output_path=results retrain threaded mpi > experiments/gmsc.log
+srun julia --project=experiments --threads 4 experiments/run_experiments.jl -- data=gmsc output_path=results retrain threaded mpi > experiments/gmsc.log

experiments/mnist.sh

@@ -2,8 +2,8 @@
 
 #SBATCH --job-name="MNIST (ECCCo)"
 #SBATCH --time=24:00:00
-#SBATCH --ntasks=48
-#SBATCH --cpus-per-task=1
+#SBATCH --ntasks=12
+#SBATCH --cpus-per-task=4
 #SBATCH --partition=compute
 #SBATCH --mem-per-cpu=4GB
 #SBATCH --account=innovation
@@ -11,4 +11,4 @@
 
 module load 2023r1 openmpi
 
-srun julia --project=experiments experiments/run_experiments.jl -- data=mnist output_path=results mpi > experiments/mnist.log
+srun julia --project=experiments --threads 4 experiments/run_experiments.jl -- data=mnist output_path=results mpi > experiments/mnist.log