[QuantizationModifier] num_calibration_steps support (PTQ) #1208
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
18 tasks
corey-nm
approved these changes
Dec 12, 2022
bfineran
force-pushed
the
quantization-refactor/freeze-bn-qat
branch
from
1ac6873 to
3ae10f2
Compare
Base automatically changed from
quantization-refactor/freeze-bn-qat
to
quantization-refactor/main
December 12, 2022 20:59
bfineran
force-pushed
the
quantization-refactor/calibration-ptq
branch
from
2435fa1 to
fcba7fd
Compare
bfineran
added a commit
that referenced
this pull request
Dec 16, 2022
bfineran
added a commit
that referenced
this pull request
Dec 19, 2022
* [QuantizationModifier] refactor base - move deprecated code to legacy file, add object routing for yaml load (#1059) * move existing ModifierQuantization and tests to legacy file * [QuantizationModifier] refactor base - move deprecated code to legacy file, add object routing for yaml load * [QuantizationModifier] pydantic classes for defining quantization schemes to generate QConfigs (#1061) * [QuantizationModifier] pydantic classes for defining quantization schemes to generate QConfigs * review response * [WIP][QuantizationModifier] base refactor flow - quantize entire module from QuantizationScheme (#1185) * [QuantizationModifier] base refactor flow - quantize entire module from QuantizationScheme * review response * testing - lifecycle + QAT application * activate qat tests * [QuantizationModifier] improved quantization flow - control of propagation with schemes and stronger testing (#1198) * [QuantizationModifier] exclude_module_types list modifier param to disable module types from quantization (#1199) * [QuantizationModifier] submodule_schemes property impl - target specific submodules by scheme (#1201) * [QuantizationModifier] submodule_schemes property impl - target specific submodules by scheme * generalize helper fn name + quality * [QuantizationModifier] module_type_schemes - override quantization scheme by layer type (#1202) * [QuantizationModifier] module_type_schemes - override quantization scheme by layer type * yaml pydoc example * [QuantizationModifier] target hardware support (#1203) * [QuantizationModifier] freeze bn stats and disable observers for QAT finetuning support (#1206) * [QuantizationModifier] num_calibration_steps support (PTQ) (#1208) * [QuantizationModifier] override params for model fuse step (#1209) * [QuantizationModifier] refactor QuantizationScheme to its own file (#1223) * [QuantizationModifier] QATWrapper support (#1226) * [QuantizationModifier] logging support (#1231) * [QuantizationModifier] logging support * fake quantize bits logging * [QuantizationModifier] potentially re-load quantization schemes on qconfig load (#1236) * [QuantizationModifier] UX refactor - submodule_overrides and ignore (#1239) * rename modifier default_scheme -> scheme * refactor set_quantization_schemes (tests passing with existing UX) * exclude_module_types -> ignore ; adds submodule exclusion * refactor submodule and module type schemes into unified submodule_overrides * [QuantizationModifier] strict mode - raise if unmatched submodules or types (#1241) * [QuantizationModifier] take ownership of add_observers_, unit test fixes (#1261) * [QuantizationModifier] take ownership of add_observers_, unit test fixes * suggestion from review - with quality override * review - suggested comment * fixes for FloatFunctional support (resnet50 broke) * [rebase] merge in changes to legacy modifier quantization
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
ported from the legacy modifier - functionality unchanged
test_plan:
yaml tests added for serialization - further testing manual for now