-
Notifications
You must be signed in to change notification settings - Fork 445
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
Test Geti integration #4188
base: develop
Are you sure you want to change the base?
Test Geti integration #4188
Conversation
dst_dir=ov_export_dir, | ||
) | ||
xml_path = ov_export_dir / "exported_model.xml" | ||
mapi_model = self.mapi_model_class.create_model(xml_path) |
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.
There's no need to specify a particular class: Model.create(...)
should work
|
||
def test_export_and_infer_openvino(self): | ||
"""Test exporting the model to OpenVINO.""" | ||
for precision in [OTXPrecisionType.FP16, OTXPrecisionType.FP32]: |
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.
Can be done via pytest parametrization for verbosity
export_format=OTXExportFormatType.OPENVINO, | ||
export_precision=OTXPrecisionType.FP32, | ||
explain=True, | ||
export_demo_package=True, |
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.
Is there something special about exporting models as a demo package? Normal export is enough to run in the scenario declared here
if self.mapi_model_class is not None: | ||
ov_optimized_dir = self.tmp_path / "ov_optimize" | ||
ov_optimized_dir.mkdir(parents=True, exist_ok=True) | ||
unzip_exportable_code( |
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.
Doesn't optimized_path
already point to the compressed model?
Summary
How to test
Checklist
License
Feel free to contact the maintainers if that's a concern.