Update --loop_count for Bert benchmark to 64

tenstorrent · May 13, 2024 · 821f5c8 · 821f5c8
1 parent 60319ae
commit 821f5c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_benchmark_tt_perf b/run_benchmark_tt_perf
@@ -14,7 +14,7 @@ echo "Running TT-Model Benchmarking on $TT_DEVICE"
 
 # BERT - Text Classification - SST-2
 if [ "$TT_DEVICE" = "wormhole" ]; then
-    python benchmark.py -d tt -m bert -c large --task text_classification -mb 64 -df Bfp8_b -mf LoFi --loop_count 128 -opt 4 --save_output
+    python benchmark.py -d tt -m bert -c large --task text_classification -mb 64 -df Bfp8_b -mf LoFi --loop_count 64 -opt 4 --save_output
 elif [ "$TT_DEVICE" = "grayskull" ]; then
     python benchmark.py -d tt -m bert -c large --task text_classification -mb 64 --loop_count 32 -df Bfp8_b -mf LoFi -opt 4 --save_output
 else