@@ -166,34 +166,49 @@ test_model_with_qnn() {
166
166
export PYTHONPATH=$EXECUTORCH_ROOT /..
167
167
168
168
EXTRA_FLAGS=" "
169
+ # Ordered by the folder name, then alphabetically by the model name
170
+ # Following models are inside examples/qualcomm/scripts folder
169
171
if [[ " ${MODEL_NAME} " == " dl3" ]]; then
170
172
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
177
177
elif [[ " ${MODEL_NAME} " == " ic3" ]]; then
178
178
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
181
181
elif [[ " ${MODEL_NAME} " == " mb" ]]; then
182
182
EXPORT_SCRIPT=mobilebert_fine_tune
183
183
EXTRA_FLAGS=" --num_epochs 1"
184
184
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
185
191
elif [[ " ${MODEL_NAME} " == " w2l" ]]; then
186
192
EXPORT_SCRIPT=wav2letter
187
193
elif [[ " ${MODEL_NAME} " == " edsr" ]]; then
188
194
EXPORT_SCRIPT=edsr
189
195
# Additional deps for edsr
190
196
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
191
202
elif [[ " ${MODEL_NAME} " == " cvt" ]]; then
192
203
EXPORT_SCRIPT=cvt
204
+ elif [[ " ${MODEL_NAME} " == " distilbert" ]]; then
205
+ EXPORT_SCRIPT=distilbert
193
206
elif [[ " ${MODEL_NAME} " == " dit" ]]; then
194
207
EXPORT_SCRIPT=dit
195
208
elif [[ " ${MODEL_NAME} " == " efficientnet" ]]; then
196
209
EXPORT_SCRIPT=efficientnet
210
+ elif [[ " ${MODEL_NAME} " == " eurobert" ]]; then
211
+ EXPORT_SCRIPT=eurobert
197
212
elif [[ " ${MODEL_NAME} " == " focalnet" ]]; then
198
213
EXPORT_SCRIPT=focalnet
199
214
elif [[ " ${MODEL_NAME} " == " mobilevit_v1" ]]; then
@@ -202,18 +217,10 @@ test_model_with_qnn() {
202
217
EXPORT_SCRIPT=mobilevit_v2
203
218
elif [[ " ${MODEL_NAME} " == " pvt" ]]; then
204
219
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
215
220
elif [[ " ${MODEL_NAME} " == " roberta" ]]; then
216
221
EXPORT_SCRIPT=roberta
222
+ elif [[ " ${MODEL_NAME} " == " swin" ]]; then
223
+ EXPORT_SCRIPT=swin_transformer
217
224
else
218
225
echo " Unsupported model $MODEL_NAME "
219
226
exit 1
0 commit comments