Riley Dulin 3be150653c [torch][ao] Add customizable loss function to NodeAccuracySummary (#136282) ... Summary: Add a customizable loss function callback to NodeAccuracySummary to allow users to pass in their own loss function. Also, fix some type errors and propagate better exception messages when unexpected tensor comparisons occur. Finally, enhance the robustness of `generate_numeric_debug_handle` in the case where it is called multiple times on the same model, by avoiding reuse of the same IDs. Test Plan: Added a test for this case in `test_numeric_debugger`. Reviewed By: jerryzh168 Differential Revision: D62898297 Pull Request resolved: https://github.com/pytorch/pytorch/pull/136282 Approved by: https://github.com/jerryzh168		2024-09-24 03:28:12 +00:00
..
ao_migration
bc	Fix failures when default is flipped for weights_only (#127627)	2024-08-16 00:22:43 +00:00
core	Add uint16 support for observer (#136238)	2024-09-18 23:52:18 +00:00
eager
fx	Fix failures when default is flipped for weights_only (#127627)	2024-08-16 00:22:43 +00:00
jit
pt2e	[torch][ao] Add customizable loss function to NodeAccuracySummary (#136282)	2024-09-24 03:28:12 +00:00
serialized
__init__.py