[CI] Replace timm_efficientdet with timm_vision_transformer in smoketest (#100106)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100106
Approved by: https://github.com/yanboliang
diff --git a/.ci/pytorch/test.sh b/.ci/pytorch/test.sh
index 38924f8..0f9e042 100755
--- a/.ci/pytorch/test.sh
+++ b/.ci/pytorch/test.sh
@@ -427,7 +427,7 @@
python benchmarks/dynamo/check_hf_bert_perf_csv.py -f "$TEST_REPORTS_DIR/inductor_training_smoketest.csv"
# Check memory compression ratio for a few models
- for test in hf_Albert timm_efficientdet; do
+ for test in hf_Albert timm_vision_transformer; do
python benchmarks/dynamo/torchbench.py --device cuda --performance --backend inductor --amp --training \
--disable-cudagraphs --batch-size-file "$(realpath benchmarks/dynamo/torchbench_models_list.txt)" \
--only $test --output "$TEST_REPORTS_DIR/inductor_training_smoketest_$test.csv"
@@ -955,7 +955,7 @@
install_torchvision
id=$((SHARD_NUMBER-1))
if [[ "${TEST_CONFIG}" == *inductor_torchbench_smoketest_perf* ]]; then
- checkout_install_torchbench hf_Bert hf_Albert timm_efficientdet
+ checkout_install_torchbench hf_Bert hf_Albert timm_vision_transformer
PYTHONPATH=$(pwd)/torchbench test_inductor_torchbench_smoketest_perf
else
checkout_install_torchbench