Skip to content

Commit 309915c

Browse files
cccclaifacebook-github-bot
authored andcommitted
Enable conv_former model in CI (pytorch#12930)
Summary: Add conv_former to CI Rollback Plan: Differential Revision: D79120806
1 parent 63ce397 commit 309915c

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

.ci/scripts/test_model.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,9 @@ test_model_with_qnn() {
207207
EXPORT_SCRIPT=dit
208208
elif [[ "${MODEL_NAME}" == "efficientnet" ]]; then
209209
EXPORT_SCRIPT=efficientnet
210+
elif [[ "${MODEL_NAME}" == "conv_former" ]]; then
211+
EXPORT_SCRIPT=conv_former
212+
EXTRA_FLAGS="--dataset imagenet-mini/val"
210213
elif [[ "${MODEL_NAME}" == "eurobert" ]]; then
211214
EXPORT_SCRIPT=eurobert
212215
elif [[ "${MODEL_NAME}" == "focalnet" ]]; then
@@ -238,7 +241,7 @@ test_model_with_qnn() {
238241
"cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin")
239242
SCRIPT_FOLDER=oss_scripts
240243
;;
241-
"albert"|"bert"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
244+
"albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
242245
pip install evaluate
243246
SCRIPT_FOLDER=oss_scripts
244247
# 16bit models will encounter op validation fail on some operations,

.github/workflows/trunk.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -568,7 +568,7 @@ jobs:
568568
strategy:
569569
matrix:
570570
dtype: [fp32]
571-
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l]
571+
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l, conv_former]
572572
fail-fast: false
573573
with:
574574
runner: linux.2xlarge

0 commit comments

Comments
 (0)