pytorch
5dcbcc6d - [Quant][fx] Fix get_default_qconfig_dict for fused modules

Commit
2 years ago
[Quant][fx] Fix get_default_qconfig_dict for fused modules Summary: Calling `prepare_fx` with `get_default_qconfig_dict` failed for models with fused modules, such as `ConvReLU2d`. This commit fixes this by adding qconfig entries for ReLU and BatchNorm as well. Test Plan: python test/test_quantization.py TestQuantizeFx.test_qconfig_dict_with_fused_modules Reviewers: jerryzh168 Subscribers: jerryzh168, vkuzo Issue: https://github.com/pytorch/pytorch/issues/75825 Pull Request resolved: https://github.com/pytorch/pytorch/pull/75838 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading