We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2e07850 commit 9367977Copy full SHA for 9367977
.github/workflows/ci-temp-model-test.yaml
@@ -37,6 +37,6 @@ jobs:
37
"""\
38
import flag_gems\
39
flag_gems.enable(record=False, once=True, path='"'"'benchmark/oplist.log'"'"')""")' benchmark/torchbenchmark/__init__.py
40
- run_command python benchmark/install.py models resnet50
41
- run_command python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json --models resnet50
+ run_command python benchmark/install.py models hf_GPT2
+ run_command python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json --models hf_GPT2
42
run_command sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py
0 commit comments