-
Notifications
You must be signed in to change notification settings - Fork 183
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
benchmarker API #1259
benchmarker API #1259
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good overall - need a way to instantiate model and pipeline from other sources (ie files, kwargs, etc)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM overall - the only thing I think would be missing is a way to pass an already instantiated pipeline object to benchmarker
something like:
my_pipeline = Pipeline.create(...)
benchmarker = Benchmarker(pipeline=my_pipeline)
Trigger
deepsparse.benchmark
deepsparse.benchmark_pipeline
Example code in tests