准备进行传播比较
- class torch.ao.quantization.prepare_for_propagation_comparison(model)[source]
向具有数值调试句柄的节点添加输出记录器
- 参数:
模型(GraphModule)- 原始模型
- 返回值:
具有所有节点输出日志器的模型,具有 numeric_debug_handle_id
- 返回类型:
向具有数值调试句柄的节点添加输出记录器
模型(GraphModule)- 原始模型
具有所有节点输出日志器的模型,具有 numeric_debug_handle_id