experimental_experiment.torch_bench.export_model¶
Export a model¶
The script export a model with different options.
python -m experimental_experiment.torch_bench.export_model --help
Example, run llama model with onnxrt backend on cuda.
python -m experimental_experiment.torch_bench.export_model \
--exporter torch_script --device cuda --config medium