准备传播对比

classtorch.ao.quantization.prepare_for_propagation_comparison(model)

为具有 numeric_debug_handle 的节点添加输出日志

参数

model (GraphModule) — 原始模型

返回值

一个具有所有具有 numeric_debug_handle_id 节点输出日志器的模型

返回类型

GraphModule