-
Notifications
You must be signed in to change notification settings - Fork 703
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
🔨 Added Tests for POT and NNCF Compression Types #2438
base: main
Are you sure you want to change the base?
🔨 Added Tests for POT and NNCF Compression Types #2438
Conversation
@ashwinvaidya17, will you be able to review this? |
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.
Thanks for the PR, I've added a few comments
…updated test_export Signed-off-by: Haroon <[email protected]>
Signed-off-by: Haroon <[email protected]>
Signed-off-by: Haroon <[email protected]>
Signed-off-by: Haroon <[email protected]>
b94c6c5
to
c4a11d5
Compare
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.
Thanks
@samet-akcay is there anything to be done in this pr ? |
📝 Description
-This PR addresses the missing tests for the POT (Post-Training Optimization Tool) and NNCF (Neural Network Compression Framework) features in Engine.export. This PR adds unit tests to verify the export functionality with these compression types.
Ensures
compression_type
is used only with OPENVINO export type, raising an error if violated in Engine.exportAdded tests in
test_cli
to cover all compression type combinations.attributes POT (Post-Training Optimization) and NNCF (Neural Network Compression Framework) to the
CompressionType
🛠️ Fixes [Task]: Add unit tests for
Engine.export
#2273✨ Changes
Select what type of change your PR is:
✅ Checklist
Before you submit your pull request, please make sure you have completed the following steps:
For more information about code review checklists, see the Code Review Checklist.