File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 4141 """\
4242 import flag_gems\
4343 flag_gems.enable(record=False, once=True, path='"'"'benchmark/oplist.log'"'"')""")' benchmark/torchbenchmark/__init__.py
44- python benchmark/install.py models hf_GPT2 resnet50 hf_T5 dcgan
45- python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output1.json --models hf_GPT2
46- python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output2.json --models resnet50
47- python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output3.json --models hf_T5
48- python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output4.json --models dcgan
44+ python benchmark/install.py
45+ python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json
4946 sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py
You can’t perform that action at this time.
0 commit comments