use inbench

Signed-off-by: zehao-intel <zehao.huang@intel.com>
intel · Jul 15, 2024 · 38fdcff · 38fdcff
1 parent 177ee53
commit 38fdcff
Showing 1 changed file with 20 additions and 8 deletions.
diff --git a/examples/3.x_api/pytorch/cv/mixed_precision/run_benchmark.sh b/examples/3.x_api/pytorch/cv/mixed_precision/run_benchmark.sh
@@ -61,14 +61,26 @@ function run_benchmark {
     else
         extra_cmd="${dataset_location}"
     fi
-
-    python main.py \
-            --pretrained \
-            --tuned_checkpoint ${tuned_checkpoint} \
-            -b ${batch_size} \
-            -a ${input_model} \
-            ${mode_cmd} \
-            ${extra_cmd}
+    if [[ ${mode} == "accuracy" ]]; then
+        python main.py \
+                --pretrained \
+                --tuned_checkpoint ${tuned_checkpoint} \
+                -b ${batch_size} \
+                -a ${input_model} \
+                ${mode_cmd} \
+                ${extra_cmd}
+    elif [[ ${mode} == "performance" ]]; then
+        incbench --num_c 4 --num_i 7 python main.py \
+                --pretrained \
+                --tuned_checkpoint ${tuned_checkpoint} \
+                -b ${batch_size} \
+                -a ${input_model} \
+                ${mode_cmd} \
+                ${extra_cmd}
+    else
+        echo "Error: No such mode: ${mode}"
+        exit 1
+    fi
 }
 
 main "$@"