diff options
| author | Přemysl Eric Janouch <p@janouch.name> | 2024-02-13 15:44:42 +0100 |
|---|---|---|
| committer | Přemysl Eric Janouch <p@janouch.name> | 2024-02-13 15:44:42 +0100 |
| commit | 19705527a0b24236af12e279f4e184bac174ba68 (patch) | |
| tree | 63b837c9bc0cf8d0259d00822f3b5a5d674ccd7f /deeptagger | |
| parent | 9e22bd0e20bc3a3ca7671ae7ab1865be5aee1f16 (diff) | |
| download | gallery-master.tar.gz gallery-master.tar.xz gallery-master.zip | |
Diffstat (limited to 'deeptagger')
| -rwxr-xr-x | deeptagger/bench.sh | 12 |
1 files changed, 5 insertions, 7 deletions
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 |