Skip to content

Commit 68e42d0

Browse files
ashwinvaidya17Ashwin Vaidya
andauthored
Add metadata to optimized model (#2618)
* bug fix for legacy openvino models * Add metadata to optimized model * Revert formatting changes --------- Co-authored-by: Ashwin Vaidya <[email protected]>
1 parent 0d9dff8 commit 68e42d0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/otx/algorithms/anomaly/tasks/openvino.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,8 @@ def optimize(
361361
output_model.optimization_type = ModelOptimizationType.POT
362362
output_model.optimization_methods = [OptimizationMethod.QUANTIZATION]
363363
output_model.precision = [ModelPrecision.INT8]
364+
metadata = self.get_metadata()
365+
output_model.set_data("metadata", json.dumps(metadata).encode())
364366

365367
self.task_environment.model = output_model
366368
self.inference_model = self.get_openvino_model()

0 commit comments

Comments
 (0)