diff --git a/experiments/experiment.jl b/experiments/experiment.jl index a6b13920a725cda35d0ed76b72a7557ddce6750c..022946bd9756269efb61e3a3340ea0b6be2aacb8 100644 --- a/experiments/experiment.jl +++ b/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." diff --git a/experiments/gmsc.sh b/experiments/gmsc.sh index 9d4f54f48eb8f89b48efa13b5c4133d9dbe1dba0..6dd647de2c32be20f836a2a64e793f0a7c340371 100644 --- a/experiments/gmsc.sh +++ b/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 diff --git a/experiments/mnist.sh b/experiments/mnist.sh index 8823002ac8addd0a41b6f90d0b7d20dfb903aa25..ebcfeb9c66e6cce7a1cc832f37ff1c2e9330dcda 100644 --- a/experiments/mnist.sh +++ b/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