diff --git a/scripts/run_benchmark.sh b/scripts/run_benchmark.sh index f45cac51556c5a3319afe1f76cf0168e3c188c04..10c3418a1b7864802a8220b4f1cbee1cd5997e99 100644 --- a/scripts/run_benchmark.sh +++ b/scripts/run_benchmark.sh @@ -3,7 +3,7 @@ set -e dim=2 for nodes in {1..1}; do - for gpus in {1,8}; do + for gpus in {1}; do for batch_size in {64,128}; do for iteration in {1..1}; do