Skip to content

Commit 8d06c79

Browse files
cccclaifacebook-github-bot
authored andcommitted
Enable fbnet model in CI
Differential Revision: D79741919
1 parent c52f6a0 commit 8d06c79

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

.ci/scripts/test_model.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,10 @@ test_model_with_qnn() {
212212
EXPORT_SCRIPT=efficientnet
213213
elif [[ "${MODEL_NAME}" == "eurobert" ]]; then
214214
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"
215219
elif [[ "${MODEL_NAME}" == "focalnet" ]]; then
216220
EXPORT_SCRIPT=focalnet
217221
elif [[ "${MODEL_NAME}" == "mobilevit_v1" ]]; then
@@ -241,7 +245,7 @@ test_model_with_qnn() {
241245
"cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin")
242246
SCRIPT_FOLDER=oss_scripts
243247
;;
244-
"albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
248+
"albert"|"bert"|"conv_former"|"distilbert"| "fbnet" | "roberta"|"efficientnet"|"mobilevit_v1")
245249
pip install evaluate
246250
SCRIPT_FOLDER=oss_scripts
247251
# 16bit models will encounter op validation fail on some operations,

0 commit comments

Comments
 (0)