diff --git a/benchmark_nnunet_pytorch_berzelius.sh b/benchmark_nnunet_pytorch_berzelius.sh new file mode 100644 index 0000000000000000000000000000000000000000..df5a074021bae09bddbb0804cad8fd531fd1a657 --- /dev/null +++ b/benchmark_nnunet_pytorch_berzelius.sh @@ -0,0 +1,22 @@ +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 + 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 + 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