Compare commits

...

1 Commits

Author SHA1 Message Date
85e05a086c only run hf_T5 2024-11-26 10:57:20 -08:00
2 changed files with 4 additions and 0 deletions

View File

@ -228,6 +228,9 @@ function checkout_install_torchbench() {
git clone https://github.com/pytorch/benchmark torchbench
pushd torchbench
git checkout "$commit"
rm -rf torchbenchmark/models/*
git checkout -- torchbenchmark/models/__init__.py
git checkout -- torchbenchmark/models/hf_T5
if [ "$1" ]; then
python install.py --continue_on_fail models "$@"

View File

@ -500,6 +500,7 @@ test_perf_for_dashboard() {
elif [[ "$target" == "accuracy" ]]; then
target_flag+=( --no-translation-validation)
fi
target_flag+=( --only hf_T5)
if [[ "$DASHBOARD_TAG" == *default-true* ]]; then
$TASKSET python "benchmarks/dynamo/$suite.py" \