Skip to content

Add max_num_iterations parameter for accuracy-aware quantization#6986

Merged
A-Artemis merged 4 commits into
developfrom
aurelien/expose-quantization-max-iteration-parameter
Jul 10, 2026
Merged

Add max_num_iterations parameter for accuracy-aware quantization#6986
A-Artemis merged 4 commits into
developfrom
aurelien/expose-quantization-max-iteration-parameter

Conversation

@A-Artemis

@A-Artemis A-Artemis commented Jul 7, 2026

Copy link
Copy Markdown
Contributor

Summary

Adds a NumberField to the quantization options for the new parameter, and accepts the new parameter on the backend.

image

Checklist

  • The PR title and description are clear and descriptive
  • I have manually tested the changes
  • All changes are covered by automated tests
  • All related issues are linked to this PR (if applicable)
  • Documentation has been updated (if applicable)

@A-Artemis A-Artemis self-assigned this Jul 7, 2026
Copilot AI review requested due to automatic review settings July 7, 2026 09:10
@A-Artemis A-Artemis requested review from a team as code owners July 7, 2026 09:10
@github-actions github-actions Bot added TEST Any changes in tests Geti Backend Issues related to the Geti application server Geti UI Issues related to Geti application frontend Geti Library Issues related to Geti Library (OTX) labels Jul 7, 2026
jpggvilaca
jpggvilaca previously approved these changes Jul 7, 2026

@jpggvilaca jpggvilaca left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

UI lgtm

@github-actions

github-actions Bot commented Jul 7, 2026

Copy link
Copy Markdown

📊 Test coverage report

Metric Coverage
Lines 60.2%
Functions 56.8%
Branches 51.8%
Statements 59.8%

@github-actions

github-actions Bot commented Jul 7, 2026

Copy link
Copy Markdown

🐳 Docker image sizes

Device Size
xpu 10801.7 MB (10.55 GB)
cpu 4013.1 MB (3.92 GB)
cuda 10208.3 MB (9.97 GB)

@codecov-commenter

Copy link
Copy Markdown

⚠️ Please install the 'codecov app svg image' to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

❌ Patch coverage is 40.00000% with 3 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
...brary/src/getitune/backend/openvino/models/base.py 0.00% 3 Missing ⚠️

📢 Thoughts on this report? Let us know!

@github-actions

github-actions Bot commented Jul 7, 2026

Copy link
Copy Markdown

⏱️ Backend import time — app.main

Accelerator Cumulative (s) Self (s)
cpu 5.082 0.004
cuda 4.121 0.003
xpu 4.624 0.003

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds support for configuring max_num_iterations for accuracy-aware quantization end-to-end (UI → backend job API → GetiTune/OpenVINO quantization), wiring it into NNCF’s advanced accuracy restorer settings and expanding test coverage across components.

Changes:

  • Library: extend OV PTQ optimization to accept max_num_iterations and translate it into NNCF AdvancedAccuracyRestorerParameters for accuracy-aware PTQ.
  • Backend: accept/propagate max_num_iterations through quantization job request/params/execution and include it in job metadata.
  • UI: add a new quantization dialog field and extend Playwright tests to validate the new parameter is submitted.

Reviewed changes

Copilot reviewed 13 out of 13 changed files in this pull request and generated 6 comments.

Show a summary per file
File Description
library/tests/unit/backend/openvino/test_engine.py Adds unit assertions for max_num_iterations handling in PTQ config.
library/src/getitune/backend/openvino/models/base.py Pops max_num_iterations and maps it to NNCF advanced accuracy restorer parameters.
library/src/getitune/backend/openvino/engine.py Extends OVEngine.optimize() API to accept and forward max_num_iterations.
application/ui/tests/models/models-page.ts Adds page object accessor for the new iterations input.
application/ui/tests/models/model-details.spec.ts Extends UI test to submit/verify max_num_iterations in quantization job payload.
application/ui/src/features/models/model-listing/model-variants/quantization-dialog/quantization-fields.component.tsx Adds the new “Max number of iterations” field and default parameter.
application/ui/src/features/models/model-listing/model-variants/quantization-dialog/quantization-dialog.component.tsx Wires the new UI state into the quantize job request body.
application/backend/tests/unit/execution/quantization/test_getitune_quantizer.py Extends unit tests to assert propagation of max_num_iterations.
application/backend/tests/unit/api/routers/test_jobs.py Extends router tests to include/validate the new parameter.
application/backend/app/models/jobs/quantization_job.py Adds max_num_iterations to quantization job params model.
application/backend/app/execution/quantization/getitune_quantizer.py Propagates max_num_iterations through quantization execution into OVEngine.
application/backend/app/api/schemas/jobs/quantization.py Adds request and metadata schema support for max_num_iterations.
application/backend/app/api/routers/jobs.py Passes the new parameter when creating quantization jobs.

Comment thread library/src/getitune/backend/openvino/engine.py Outdated
Comment thread application/backend/app/execution/quantization/getitune_quantizer.py Outdated
Comment thread application/backend/app/api/schemas/jobs/quantization.py
Comment thread library/src/getitune/backend/openvino/models/base.py
@A-Artemis A-Artemis enabled auto-merge July 7, 2026 13:07
@A-Artemis A-Artemis added this pull request to the merge queue Jul 9, 2026
@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to failed status checks Jul 9, 2026
@A-Artemis A-Artemis enabled auto-merge July 10, 2026 05:43
@A-Artemis A-Artemis added this pull request to the merge queue Jul 10, 2026
Merged via the queue into develop with commit 6e2b51b Jul 10, 2026
56 of 73 checks passed
@A-Artemis A-Artemis deleted the aurelien/expose-quantization-max-iteration-parameter branch July 10, 2026 07:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Geti Backend Issues related to the Geti application server Geti Library Issues related to Geti Library (OTX) Geti UI Issues related to Geti application frontend TEST Any changes in tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Expose parameter to configure max iterations in accuracy-aware quantization

6 participants