Ean Garvey 3514822cac Improvements to pytest benchmarks. (#267) ...

* Add ONNX env var flags for venv setup.

* Setup arguments for ONNX benchmarking via pytest.

* Enable ONNX benchmarking on MiniLM via pytest (experimental)

* Fix sequence lengths to 128 for TF model creation and fix issue with benchmarks.

* Disable CI CPU benchmarks on A100, change some default args.

* add xfails for roberta TF model tests on GPU.

2022-08-17 02:29:48 -05:00

3.8 KiB

Raw Permalink Blame History

View Raw