快捷方式

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)

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

獲取針對初學者和高級開發者的深入教學

檢視教學

資源

尋找開發資源並獲取您的問題解答

檢視資源