From 619f012fe6eafbec32c6dbde23ebadbf3437fe1d Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Tue, 30 May 2023 19:46:32 +0000
Subject: [PATCH] Update file run_benchmark.sh
---
scripts/run_benchmark.sh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/scripts/run_benchmark.sh b/scripts/run_benchmark.sh
index c669cad..f45cac5 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..1}; do
+ for gpus in {1,8}; do
for batch_size in {64,128}; do
for iteration in {1..1}; do
@@ -27,7 +27,7 @@ done
dim=3
for nodes in {1..1}; do
- for gpus in {1..1}; do
+ for gpus in {1,8}; do
for batch_size in {1,2,4}; do
for iteration in {1..1}; do
--
GitLab