-
-
Notifications
You must be signed in to change notification settings - Fork 110
Open
Labels
featureNew feature or requestNew feature or request
Description
Problem
When writing custom models or pipelines, one may want to react to specific inference events, for example before/after benchmarking on a file.
Idea
Add classes RealTimeInferenceHook and BenchmarkHook to define listener interfaces. This can also be used to implement other behavior like progress bars, writing results, etc.
Example
class MyHook(BenchmarkHook):
def on_finished(results):
print("Finished!")
benchmark = Benchmark(..., hooks=[MyHook()])Metadata
Metadata
Assignees
Labels
featureNew feature or requestNew feature or request