fix: Update dataset location for resnet models

Signed-off-by: Sun, Xuehao <[email protected]>
onnx · Jul 1, 2024 · 7a5e0ee · 7a5e0ee
1 parent 4418482
commit 7a5e0ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.azure-pipelines/scripts/models/run_onnxrt_models_trigger.sh b/.azure-pipelines/scripts/models/run_onnxrt_models_trigger.sh
@@ -20,7 +20,7 @@ done
 
 CONFIG_PATH="/neural-compressor/examples/.config/model_params_onnxrt.json"
 model_src_dir=$(jq -r ".\"onnxrt\".\"$model\".\"model_src_dir\"" "$CONFIG_PATH")
-if [ $model == *"resnet"* ]; then
+if [[ "$model" == *"resnet"* ]]; then
     dataset_location="/tf_dataset2/datasets/imagenet/ImagenetRaw/ImagenetRaw_small_5000/ILSVRC2012_img_val"
     label_path="/tf_dataset2/datasets/imagenet/ImagenetRaw/ImagenetRaw_small_5000/val.txt"
 else