From 8e25315ad3f42f318faf8223ccc2122d720cabde Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuagu37@gmail.com>
Date: Mon, 10 Oct 2022 14:15:38 +0200
Subject: [PATCH] Create benchmark_nnunet_pytorch_berzelius.sh
---
benchmark_nnunet_pytorch_berzelius.sh | 22 ++++++++++++++++++++++
1 file changed, 22 insertions(+)
create mode 100644 benchmark_nnunet_pytorch_berzelius.sh
diff --git a/benchmark_nnunet_pytorch_berzelius.sh b/benchmark_nnunet_pytorch_berzelius.sh
new file mode 100644
index 0000000..df5a074
--- /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
--
GitLab