benchmark
69723a96 - Adds the sweep runner script (#742)

Commit
3 years ago
Adds the sweep runner script (#742) Summary: This PR adds a new `run_sweep.py` script to TorchBench. It runs all specified model tests in subprocess worker, and apply optional arguments to them. Currently, only batch size is supported, we will support `fx2trt` and `torch_tensorrt` in a follow-up PR. Pull Request resolved: https://github.com/pytorch/benchmark/pull/742 Reviewed By: erichan1 Differential Revision: D34134739 Pulled By: xuzhao9 fbshipit-source-id: f798f415f976d7e3deee7ffed35f4cbfdb7487f0
Author
Parents
Loading