From 9671ec3c3394b59aae736669ce9fa1c8fc18be4a Mon Sep 17 00:00:00 2001
From: evonneng <evonne_ng@berkeley.edu>
Date: Wed, 11 May 2022 17:59:35 +0000
Subject: [PATCH] update readme

---
 README.md                                    |  6 ++++++
 scripts/benchmarking/launch_train_blender.sh | 14 ++++++++++++++
 2 files changed, 20 insertions(+)
 create mode 100755 scripts/benchmarking/launch_train_blender.sh

diff --git a/README.md b/README.md
index bcf1dac65..03f34916c 100644
--- a/README.md
+++ b/README.md
@@ -105,3 +105,9 @@ pip install py-spy
 ```
 ipython --pdb scripts/run_train.py
 ```
+
+5. **Benchmarking** For launching training jobs automatically on blender dataset
+
+```
+./scripts/benchmarking/launch_train_blender.sh <gpu0> <gpu1> ... <gpu7>
+```
\ No newline at end of file
diff --git a/scripts/benchmarking/launch_train_blender.sh b/scripts/benchmarking/launch_train_blender.sh
new file mode 100755
index 000000000..31165ea4c
--- /dev/null
+++ b/scripts/benchmarking/launch_train_blender.sh
@@ -0,0 +1,14 @@
+#!/bin/bash
+
+DATASETS=("mic" "ficus" "chair" "hotdog" "materials" "drums" "ship" "lego")
+GPU_IDX=1
+for dataset in ${DATASETS[@]}; do
+    export CUDA_VISIBLE_DEVICES=${!GPU_IDX}
+    python scripts/run_train.py \
+           data.dataset.data_directory=data/blender/${dataset} \
+           experiment_name=blender_${dataset} \
+           logging.enable_stats=False \
+           logging.enable_profiler=False &
+    echo "Launched ${dataset} on gpu ${!GPU_IDX}"
+    GPU_IDX=$((GPU_IDX+1))
+done
\ No newline at end of file