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 的節點都有輸出記錄器