-
Notifications
You must be signed in to change notification settings - Fork 268
Support models without a batch dimension #3721
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
base: develop
Are you sure you want to change the base?
Support models without a batch dimension #3721
Conversation
|
@daniil-lyakhov , please take a look |
daniil-lyakhov
left a comment
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.
@Thakor-Yashpal, thank you for the contribution! Please apply the comments and add the test reproducer to the PR
.gitignore
Outdated
| .installed.cfg | ||
| *.egg | ||
|
|
||
| my_project_env |
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.
| my_project_env |
src/nncf/version.py
Outdated
| # limitations under the License. | ||
|
|
||
| __version__ = "2.19.0" | ||
| __version__ = "2.19.0.dev0+1489448a3dirty" |
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.
| __version__ = "2.19.0.dev0+1489448a3dirty" | |
| __version__ = "2.19.0" |
| # Copyright (c) 2025 Intel Corporation | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at |
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.
Looks like some issues with the code merging, there are two different classes with the same name in the PR
Changes
nncf/quantization/algorithms/fast_bias_correction/openvino_backend.py.create_input_datamethod to add a specific check for 1D input shapes (i.e.,len(shape) == 1).datalist.Reason for changes
The
create_input_datamethod was looping based on the channel dimension size (3) but should have been based on the statistics list size (1)Related tickets
Closes the feature request: "Currently FastBiasCorrection algorithm does not support models without the batch dimension"
#3481
Tests
Verified the fix by running the reproducer command provided in the issue, which now passes: