@@ -166,28 +166,30 @@ test_model_with_qnn() {
166166 export PYTHONPATH=$EXECUTORCH_ROOT /..
167167
168168 EXTRA_FLAGS=" "
169+ # Following models are inside examples/qualcomm/scripts folder
169170 if [[ " ${MODEL_NAME} " == " dl3" ]]; then
170171 EXPORT_SCRIPT=deeplab_v3
171- elif [[ " ${MODEL_NAME} " == " mv3" ]]; then
172- EXPORT_SCRIPT=mobilenet_v3
173- elif [[ " ${MODEL_NAME} " == " mv2" ]]; then
174- EXPORT_SCRIPT=mobilenet_v2
175- elif [[ " ${MODEL_NAME} " == " ic4" ]]; then
176- EXPORT_SCRIPT=inception_v4
172+ elif [[ " ${MODEL_NAME} " == " edsr" ]]; then
173+ EXPORT_SCRIPT=edsr
174+ # Additional deps for edsr
175+ pip install piq
177176 elif [[ " ${MODEL_NAME} " == " ic3" ]]; then
178177 EXPORT_SCRIPT=inception_v3
179- elif [[ " ${MODEL_NAME} " == " vit " ]]; then
180- EXPORT_SCRIPT=torchvision_vit
178+ elif [[ " ${MODEL_NAME} " == " ic4 " ]]; then
179+ EXPORT_SCRIPT=inception_v4
181180 elif [[ " ${MODEL_NAME} " == " mb" ]]; then
182181 EXPORT_SCRIPT=mobilebert_fine_tune
183182 EXTRA_FLAGS=" --num_epochs 1"
184183 pip install scikit-learn
184+ elif [[ " ${MODEL_NAME} " == " mv2" ]]; then
185+ EXPORT_SCRIPT=mobilenet_v2
186+ elif [[ " ${MODEL_NAME} " == " mv3" ]]; then
187+ EXPORT_SCRIPT=mobilenet_v3
188+ elif [[ " ${MODEL_NAME} " == " vit" ]]; then
189+ EXPORT_SCRIPT=torchvision_vit
185190 elif [[ " ${MODEL_NAME} " == " w2l" ]]; then
186191 EXPORT_SCRIPT=wav2letter
187- elif [[ " ${MODEL_NAME} " == " edsr" ]]; then
188- EXPORT_SCRIPT=edsr
189- # Additional deps for edsr
190- pip install piq
192+ # Following models are inside examples/qualcomm/oss_scripts folder
191193 elif [[ " ${MODEL_NAME} " == " albert" ]]; then
192194 EXPORT_SCRIPT=albert
193195 elif [[ " ${MODEL_NAME} " == " bert" ]]; then
0 commit comments