benchmark: fix main benchmark script

dbmdz · Mar 1, 2018 · a15b238 · a15b238
1 parent a630a1f
commit a15b238
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/benchmark_all.sh b/benchmark_all.sh
@@ -1,6 +1,11 @@
-ARCHITECTURES="lstm"
+LANGUAGES=$(ls data/*.sentences.train)
+ARCHITECTURES="lstm bi-lstm cnn"
 
 for architecture in $ARCHITECTURES
 do
-  ./benchmark_language.sh $architecture
+  for train_file in $LANGUAGES
+  do
+    lang=${train_file%.train}
+    ./benchmark_language.sh $lang $architecture
+  done
 done