Source code for experimental_experiment.torch_bench.bash_bench_torchbench
"""Benchmark exporters===================Benchmarks many models from the `HuggingFace <https://huggingface.co/models>`_.Available exporters:* eager: identity* export: :func:`torch.export.export`* compile: :func:`torch.compile`* custom: :func:`experimental_experiment.torch_interpreter.to_onnx`* script: :func:`torch.onnx.export`* onnx_dynamo: :func:`torch.onnx.export` with ``dynamo=True``:: python -m experimental_experiment.torch_bench.bash_bench_torchbench --help:: python -m experimental_experiment.torch_bench.bash_bench_torchbench --model ""Extra dependencies:* https://github.com/pytorch/benchmark* effdet* fbgemm_gpu_nightly* iopath* opencv-python* pyre-extensions* torchrec"""fromexperimental_experiment.torch_bench._bash_bench_cmdimportbash_bench_main
[docs]defmain(args=None):""" Main function for command line ``python -m experimental_experiment.torch_bench.bash_bench_torchbench``. """bash_bench_main("bash_bench_torchbench",__doc__,args)