对比结果

classtorch.ao.quantization.compare_results(ref_results, actual_results)

给定两个从debug_handle_id(整数)到张量列表的映射,返回一个从debug_handle_idNodeAccuracySummary的映射,并包含如SQNR、MSE等比较信息。

参数
  • ref_results (Dict[int, Tuple[str, object, List[torch.Tensor]]]) – 每个 debug_handle_id 对应的参考结果

  • actual_results (Dict[int, Tuple[str, object, List[torch.Tensor]]]) – 每个 debug_handle_id 对应的实际结果

返回值

字典[int, NodeAccuracySummary]

返回类型

Dict[int, NodeAccuracySummary]

本页目录