mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
Fixes #144522 ## Description FX QAT docs for prepare_qat_fx incorrectly used get_default_qat_qconfig when it should use get_default_qat_qconfig_mapping for a qconfig_mapping. Previous example code incorrectly used `get_default_qat_qconfig`, resulting in a qconfig being incorrectly passed to `prepare_qat_fx`. `prepare_qat_fx` requires a `qconfig_mapping`, not a single `qconfig`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/155100 Approved by: https://github.com/jerryzh168 |
||
|---|---|---|
| .. | ||
| nn | ||
| ns | ||
| pruning | ||
| quantization | ||
| __init__.py | ||