快捷方式

generate_numeric_debug_handle

class torch.ao.quantization.generate_numeric_debug_handle(ep)[source]

为给定 ExportedProgram 的图模块中的所有节点(如 conv2d、squeeze、conv1d 等)附加 numeric_debug_handle_id,除了 placeholder。请注意,像 getattr 这样的节点不在范围内,因为它们不在图中。

输入的导出程序的图节点会被就地修改。

这是一个使用调试句柄量化流程的示例

ep = export_for_training(eager_model, example_inputs)
generate_numeric_debug_handle(ep)

m = ep.module()
quantizer = XNNPACKQuantizer()
m = prepare_pt2e(m, quantizer)
m = convert_pt2e(m)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源