frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ao] Added ModelReport class outline for Fx Graph Modules

Browse Source 
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 init method and the signatures and docs for each
of the proposed helper functions.

This also address and fixes a revert issue.

Test Plan: python test/test_quantization.py TestFxModelReportClass

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/80052

Approved by: https://github.com/HDCharles
This commit is contained in:
vspenubarthi
2022-06-22 10:59:18 -07:00
committed by PyTorch MergeBot
parent d43e6c9f4a
commit 01720ae3b6
4 changed files with 384 additions and 242 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_quantization.py
View File

@ -87,6 +87,7 @@ try:
from quantization.fx.test_model_report_fx import TestFxModelReportDetector # noqa: F401
from quantization.fx.test_model_report_fx import TestFxModelReportObserver # noqa: F401
from quantization.fx.test_model_report_fx import TestFxModelReportDetectDynamicStatic # noqa: F401
from quantization.fx.test_model_report_fx import TestFxModelReportClass # noqa: F401
except ImportError:
pass