diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh new file mode 100644 index 0000000000000000000000000000000000000000..7bdcdc764745b95108b975771db0c319044842a2 --- /dev/null +++ b/scripts/benchmark_sbatch_submit.sh @@ -0,0 +1,26 @@ +cd /proj/nsc/users/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet + +dim=2 +nodes=1 + +for gpus in 8; do + + for batch_size in 256; do + + for iteration in {1..10}; do + + echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration} + + # For single node + sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration} + + # For multi node + #sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration} + + sleep 1 # pause to be kind to the scheduler + + done + + done + +done