prepare_for_propagation_comparison

class torch.ao.quantization.prepare_for_propagation_comparison(model)

為具有 numeric_debug_handle 的節點新增輸出記錄器

引數

model (GraphModule) – 原始模型

返回

一個為所有具有 numeric_debug_handle_id 的節點新增輸出記錄器的模型

返回型別

GraphModule