diff --git a/deeptagger/bench.sh b/deeptagger/bench.sh index 6b62791..d512ff0 100755 --- a/deeptagger/bench.sh +++ b/deeptagger/bench.sh @@ -28,11 +28,9 @@ run() { for model in models/*.model do name=$(sed -n 's/^name=//p' "$model") - run "" 1 "$model" "$@" - run "" 4 "$model" "$@" - run "" 16 "$model" "$@" - - run --cpu 1 "$model" "$@" - run --cpu 4 "$model" "$@" - run --cpu 16 "$model" "$@" + for batch in 1 4 16 + do + run "" $batch "$model" "$@" + run --cpu $batch "$model" "$@" + done done