microsoft / Cream

This is a collection of our NAS and Vision Transformer work.
MIT License
1.62k stars 220 forks source link

EfficientViT: benchmark code for onnx #177

Closed jameslahm closed 1 year ago

jameslahm commented 1 year ago

Thanks for your great work! Would you mind sharing the benchmark code for onnx? Thanks!

xinyuliu-jeffrey commented 1 year ago

Hi @jameslahm ,

Thanks for your interest in our work!

The part of onnx throughput benchmarking is included in some private projects, so we may not be able to release it directly at present. However, you may implement an onnx throughput inference script by referencing official onnxruntime doc and speed_test jointly. Please feel free to contact me if there are any other questions. Thanks!

Best, Xinyu

jameslahm commented 1 year ago

Thanks!