@@ -166,34 +166,49 @@ test_model_with_qnn() {
166166 export PYTHONPATH=$EXECUTORCH_ROOT /..
167167
168168 EXTRA_FLAGS=" "
169+ # Ordered by the folder name, then alphabetically by the model name
170+ # Following models are inside examples/qualcomm/scripts folder
169171 if [[ " ${MODEL_NAME} " == " dl3" ]]; then
170172 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
173+ elif [[ " ${MODEL_NAME} " == " edsr" ]]; then
174+ EXPORT_SCRIPT=edsr
175+ # Additional deps for edsr
176+ pip install piq
177177 elif [[ " ${MODEL_NAME} " == " ic3" ]]; then
178178 EXPORT_SCRIPT=inception_v3
179- elif [[ " ${MODEL_NAME} " == " vit " ]]; then
180- EXPORT_SCRIPT=torchvision_vit
179+ elif [[ " ${MODEL_NAME} " == " ic4 " ]]; then
180+ EXPORT_SCRIPT=inception_v4
181181 elif [[ " ${MODEL_NAME} " == " mb" ]]; then
182182 EXPORT_SCRIPT=mobilebert_fine_tune
183183 EXTRA_FLAGS=" --num_epochs 1"
184184 pip install scikit-learn
185+ elif [[ " ${MODEL_NAME} " == " mv2" ]]; then
186+ EXPORT_SCRIPT=mobilenet_v2
187+ elif [[ " ${MODEL_NAME} " == " mv3" ]]; then
188+ EXPORT_SCRIPT=mobilenet_v3
189+ elif [[ " ${MODEL_NAME} " == " vit" ]]; then
190+ EXPORT_SCRIPT=torchvision_vit
185191 elif [[ " ${MODEL_NAME} " == " w2l" ]]; then
186192 EXPORT_SCRIPT=wav2letter
187193 elif [[ " ${MODEL_NAME} " == " edsr" ]]; then
188194 EXPORT_SCRIPT=edsr
189195 # Additional deps for edsr
190196 pip install piq
197+ # Following models are inside examples/qualcomm/oss_scripts folder
198+ elif [[ " ${MODEL_NAME} " == " albert" ]]; then
199+ EXPORT_SCRIPT=albert
200+ elif [[ " ${MODEL_NAME} " == " bert" ]]; then
201+ EXPORT_SCRIPT=bert
191202 elif [[ " ${MODEL_NAME} " == " cvt" ]]; then
192203 EXPORT_SCRIPT=cvt
204+ elif [[ " ${MODEL_NAME} " == " distilbert" ]]; then
205+ EXPORT_SCRIPT=distilbert
193206 elif [[ " ${MODEL_NAME} " == " dit" ]]; then
194207 EXPORT_SCRIPT=dit
195208 elif [[ " ${MODEL_NAME} " == " efficientnet" ]]; then
196209 EXPORT_SCRIPT=efficientnet
210+ elif [[ " ${MODEL_NAME} " == " eurobert" ]]; then
211+ EXPORT_SCRIPT=eurobert
197212 elif [[ " ${MODEL_NAME} " == " focalnet" ]]; then
198213 EXPORT_SCRIPT=focalnet
199214 elif [[ " ${MODEL_NAME} " == " mobilevit_v1" ]]; then
@@ -202,18 +217,10 @@ test_model_with_qnn() {
202217 EXPORT_SCRIPT=mobilevit_v2
203218 elif [[ " ${MODEL_NAME} " == " pvt" ]]; then
204219 EXPORT_SCRIPT=pvt
205- elif [[ " ${MODEL_NAME} " == " swin" ]]; then
206- EXPORT_SCRIPT=swin_transformer
207- elif [[ " ${MODEL_NAME} " == " albert" ]]; then
208- EXPORT_SCRIPT=albert
209- elif [[ " ${MODEL_NAME} " == " bert" ]]; then
210- EXPORT_SCRIPT=bert
211- elif [[ " ${MODEL_NAME} " == " distilbert" ]]; then
212- EXPORT_SCRIPT=distilbert
213- elif [[ " ${MODEL_NAME} " == " eurobert" ]]; then
214- EXPORT_SCRIPT=eurobert
215220 elif [[ " ${MODEL_NAME} " == " roberta" ]]; then
216221 EXPORT_SCRIPT=roberta
222+ elif [[ " ${MODEL_NAME} " == " swin" ]]; then
223+ EXPORT_SCRIPT=swin_transformer
217224 else
218225 echo " Unsupported model $MODEL_NAME "
219226 exit 1
0 commit comments