prepare_for_propagation_comparison¶
- class torch.ao.quantization.prepare_for_propagation_comparison(model)[原始碼]¶
向具有 numeric_debug_handle 的節點新增輸出記錄器
- 引數
model (GraphModule) – 原始模型
- 返回
返回一個模型,該模型包含所有具有 numeric_debug_handle_id 的節點的輸出記錄器
- 返回型別
向具有 numeric_debug_handle 的節點新增輸出記錄器
model (GraphModule) – 原始模型
返回一個模型,該模型包含所有具有 numeric_debug_handle_id 的節點的輸出記錄器