diff --git a/experiments/mnist.sh b/experiments/mnist.sh index 14faad1b5c4c0331aa70aac9dca994924cf06246..8823002ac8addd0a41b6f90d0b7d20dfb903aa25 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=12 -#SBATCH --cpus-per-task=4 +#SBATCH --ntasks=48 +#SBATCH --cpus-per-task=1 #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 retrain output_path=results threaded mpi > experiments/mnist.log +srun julia --project=experiments experiments/run_experiments.jl -- data=mnist output_path=results mpi > experiments/mnist.log diff --git a/experiments/synthetic.sh b/experiments/synthetic.sh index f1f4b4f10ffd7941565998d40dd14db643c853cb..34b342d40a05e84051142f7920fa84b8bce2e3bf 100644 --- a/experiments/synthetic.sh +++ b/experiments/synthetic.sh @@ -11,4 +11,4 @@ module load 2023r1 openmpi -srun julia --project=experiments --threads 4 experiments/run_experiments.jl -- data=linearly_separable,moons,circles output_path=results retrain threaded mpi > experiments/synthetic.log +srun julia --project=experiments --threads 4 experiments/run_experiments.jl -- data=linearly_separable,moons,circles output_path=results threaded mpi > experiments/synthetic.log