快捷方式

generate_numeric_debug_handle

class torch.ao.quantization.generate_numeric_debug_handle(ep)[source]

為給定 ExportedProgram 的圖模組中的所有節點附加 numeric_debug_handle_id,例如 conv2d、squeeze、conv1d 等,但不包括 placeholder。請注意,getattr 等節點不在範圍之內,因為它們不在圖中。

輸入 exported program 的圖節點會被就地修改。

這是一個使用除錯控制代碼量化流程的示例

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 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源