Skip to content

Commit ae99c04

Browse files
cccclaifacebook-github-bot
authored andcommitted
Re-order the model list in CI alphabetical order (pytorch#12929)
Summary: Re-order the model in the list in alphabetical order so it's easy to track Differential Revision: D79118824
1 parent 9d92ff8 commit ae99c04

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

.ci/scripts/test_model.sh

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -166,28 +166,30 @@ test_model_with_qnn() {
166166
export PYTHONPATH=$EXECUTORCH_ROOT/..
167167

168168
EXTRA_FLAGS=""
169+
# Following models are inside examples/qualcomm/scripts folder
169170
if [[ "${MODEL_NAME}" == "dl3" ]]; then
170171
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
172+
elif [[ "${MODEL_NAME}" == "edsr" ]]; then
173+
EXPORT_SCRIPT=edsr
174+
# Additional deps for edsr
175+
pip install piq
177176
elif [[ "${MODEL_NAME}" == "ic3" ]]; then
178177
EXPORT_SCRIPT=inception_v3
179-
elif [[ "${MODEL_NAME}" == "vit" ]]; then
180-
EXPORT_SCRIPT=torchvision_vit
178+
elif [[ "${MODEL_NAME}" == "ic4" ]]; then
179+
EXPORT_SCRIPT=inception_v4
181180
elif [[ "${MODEL_NAME}" == "mb" ]]; then
182181
EXPORT_SCRIPT=mobilebert_fine_tune
183182
EXTRA_FLAGS="--num_epochs 1"
184183
pip install scikit-learn
184+
elif [[ "${MODEL_NAME}" == "mv2" ]]; then
185+
EXPORT_SCRIPT=mobilenet_v2
186+
elif [[ "${MODEL_NAME}" == "mv3" ]]; then
187+
EXPORT_SCRIPT=mobilenet_v3
188+
elif [[ "${MODEL_NAME}" == "vit" ]]; then
189+
EXPORT_SCRIPT=torchvision_vit
185190
elif [[ "${MODEL_NAME}" == "w2l" ]]; then
186191
EXPORT_SCRIPT=wav2letter
187-
elif [[ "${MODEL_NAME}" == "edsr" ]]; then
188-
EXPORT_SCRIPT=edsr
189-
# Additional deps for edsr
190-
pip install piq
192+
# Following models are inside examples/qualcomm/oss_scripts folder
191193
elif [[ "${MODEL_NAME}" == "albert" ]]; then
192194
EXPORT_SCRIPT=albert
193195
elif [[ "${MODEL_NAME}" == "bert" ]]; then

0 commit comments

Comments
 (0)