diff --git a/benchmark_nnunet_pytorch_berzelius.sh b/benchmark_nnunet_pytorch_berzelius.sh deleted file mode 100644 index 3f0e00ad9c40648ad78ab2093c6d8bc8ba088d96..0000000000000000000000000000000000000000 --- a/benchmark_nnunet_pytorch_berzelius.sh +++ /dev/null @@ -1,31 +0,0 @@ -#!/bin/bash - -#SBATCH -A nsc -#SBATCH --nodes=1 -#SBATCH --gres=gpu:8 -#SBATCH --time=0-01:00:00 -#SBATCH --reservation=your-reservation -#SBATCH -o benchmark.out - -dim=2 -nodes=1 - -for gpus in 1 2 3 4 5 6 7 8; do - - for batch_size in 1 2 4 8 16 32 64 128; do - - for iteration in {1..100}; do - - echo dim $dim, nodes $nodes, gpus $gpus, batch_size $batch_size, tf32, iteration $iteration - rm -f results/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_tf32_iteration${iteration}.json - singularity exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidia_nnu-net_for_pytorch.sif python scripts/benchmark.py --mode train --gpus $gpus --dim $dim --batch_size $batch_size --nodes $nodes --logname="benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_tf32_iteration${iteration}.json" - - echo dim $dim, nodes $nodes, gpus $gpus, batch_size $batch_size, amp, iteration $iteration - rm -f results/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_amp_iteration${iteration}.json - singularity exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidia_nnu-net_for_pytorch.sif python scripts/benchmark.py --mode train --gpus $gpus --dim $dim --batch_size $batch_size --nodes $nodes --amp --logname="benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_amp_iteration${iteration}.json" - - done - - done - -done