pytorch
1cff4147 - [ao] Added insertion of observer capability to ModelReport class

Commit
2 years ago
[ao] Added insertion of observer capability to ModelReport class Summary: The ModelReport class in model_report.py combines the functionality of the detectors and the ModelReportObserver. It creates an end-to-end system where a user can pass in a prepared Graph Model to insert the ModelReportObservers, then after the user callibrates their model, the callibrated model can then be used by the ModelReport class to generate reports based on what the user wished to gather information on. This contains the implementation and tests for the prepare_detailed_calibration method which is used on a prepared fx model to insert the desired observers for the different detectors. Test Plan: python test/test_quantization.py TestFxModelReportClass Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/79752 Approved by: https://github.com/HDCharles
Author
Committer
Parents
Loading