Fix pytorch profile and add tvm-tflite

ctuning · Oct 3, 2023 · 15e673d · 15e673d
1 parent 3c7ec89
commit 15e673d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vision/classification_and_detection/run_common.sh b/vision/classification_and_detection/run_common.sh
@@ -18,7 +18,7 @@ device="cpu"
 
 for i in $* ; do
     case $i in
-       tf|onnxruntime|tflite|pytorch|tvm-onnx|tvm-pytorch|ncnn) backend=$i; shift;;
+       tf|onnxruntime|tflite|pytorch|tvm-onnx|tvm-pytorch|tvm-tflite|ncnn) backend=$i; shift;;
        cpu|gpu) device=$i; shift;;
        gpu) device=gpu; shift;;
        resnet50|mobilenet|ssd-mobilenet|ssd-resnet34|ssd-resnet34-tf|retinanet) model=$i; shift;;
@@ -133,7 +133,7 @@ fi
 #
 if [ $name == "resnet50-tvm-pytorch" ] ; then
     model_path="$MODEL_DIR/resnet50_INT8bit_quantized.pt"
-    profile=resnet50-onnxruntime
+    profile=resnet50-pytorch
     extra_args="$extra_args --backend tvm"
 fi