generate_numeric_debug_handle¶
- class torch.ao.quantization.generate_numeric_debug_handle(ep)[原始碼]¶
針對給定的 ExportedProgram 中圖模組裡的所有節點,例如 conv2d、squeeze、conv1d 等,附加 numeric_debug_handle_id,placeholder 節點除外。請注意,getattr 之類的節點不在圖模組中,因此不在範圍內。
輸入的 ExportedProgram 的圖節點會被就地修改。
以下是使用 debug handle 量化流程的範例
ep = export_for_training(eager_model, example_inputs) generate_numeric_debug_handle(ep) m = ep.module() quantizer = XNNPACKQuantizer() m = prepare_pt2e(m, quantizer) m = convert_pt2e(m)