Fixed small error in asha script, added some progress indication to t…

…he asha and sha benchmarking scripts
AbhinavBehal · Jun 1, 2019 · 74cec23 · 74cec23
1 parent 0defc4c
commit 74cec23
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/benchmarking/asha.sh b/benchmarking/asha.sh
@@ -9,13 +9,15 @@ mkdir -p "${current_folder}/results" && touch ${results_file}
 echo "Workers,Min_R,Max_R,Reduction_Factor,Early_Stopping_Rounds,Time,Score" > ${results_file}
 
 for i in 2 4 8 16 32 64 128 256 512 1024; do
+    echo ${i}
+
     workers=50
     min_r=1
     max_r=${i}
     early_stopping_rounds=${i}
     reduction_factor=2
 
-    params="{\"n_workers\": ${i}, \"min_r\": ${min_r}, \"max_r\": ${max_r},\
+    params="{\"n_workers\": ${workers}, \"min_r\": ${min_r}, \"max_r\": ${max_r},\
      \"reduction_factor\": ${reduction_factor}, \"early_stopping_rounds\": ${early_stopping_rounds},\"cv\": 3}"
 
     python "${current_folder}/../main.py" -a asha -p "${params}" > ${temp_file}

diff --git a/benchmarking/sha.sh b/benchmarking/sha.sh
@@ -9,6 +9,8 @@ mkdir -p "${current_folder}/results" && touch ${results_file}
 echo "Configs,Min_R,Max_R,Reduction_Factor,Time,Score" > ${results_file}
 
 for i in 2 4 8 16 32 64 128 256 512 1024; do
+    echo ${i}
+
     min_r=1
     max_r=${i}
     reduction_factor=2