File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -212,6 +212,10 @@ test_model_with_qnn() {
212
212
EXPORT_SCRIPT=efficientnet
213
213
elif [[ " ${MODEL_NAME} " == " eurobert" ]]; then
214
214
EXPORT_SCRIPT=eurobert
215
+ elif [[ " ${MODEL_NAME} " == " fbnet" ]]; then
216
+ pip install timm==0.1.24
217
+ EXPORT_SCRIPT=fbnet
218
+ EXTRA_FLAGS=" --dataset imagenet-mini/val"
215
219
elif [[ " ${MODEL_NAME} " == " focalnet" ]]; then
216
220
EXPORT_SCRIPT=focalnet
217
221
elif [[ " ${MODEL_NAME} " == " mobilevit_v1" ]]; then
@@ -241,7 +245,7 @@ test_model_with_qnn() {
241
245
" cvt" |" dit" |" focalnet" |" mobilevit_v2" |" pvt" |" swin" )
242
246
SCRIPT_FOLDER=oss_scripts
243
247
;;
244
- " albert" |" bert" |" conv_former" |" distilbert" |" roberta" |" efficientnet" |" mobilevit_v1" )
248
+ " albert" |" bert" |" conv_former" |" distilbert" | " fbnet " | " roberta" |" efficientnet" |" mobilevit_v1" )
245
249
pip install evaluate
246
250
SCRIPT_FOLDER=oss_scripts
247
251
# 16bit models will encounter op validation fail on some operations,
You can’t perform that action at this time.
0 commit comments