Skip to content
Snippets Groups Projects
Unverified Commit a49880a3 authored by Xuan Gu's avatar Xuan Gu Committed by GitHub
Browse files

Update benchmark_multi_node.sh

parent 10e1f253
No related branches found
No related tags found
No related merge requests found
...@@ -16,7 +16,7 @@ srun singularity exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidi ...@@ -16,7 +16,7 @@ srun singularity exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidi
# For enroot # For enroot
#rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_tf32_iteration${5}.json #rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_tf32_iteration${5}.json
#srun enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet sh -c 'python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --logname="benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_tf32_iteration${5}.json"' #srun enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet sh -c "python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --logname="benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_tf32_iteration${5}.json""
#rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json #rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json
#srun enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet sh -c 'python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --amp --logname="benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json"' #srun enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet sh -c "python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --amp --logname="benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json""
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment