.torch_bench.export_model

Export a model

The script export a model with different options.

python -m experimental_experiment.torch_bench.export_model --help

python -m experimental_experiment.torch_bench.export_model \
       --exporter torch_script --device cuda --config medium