后处理函数

模块: polygraphy.tools.args

class ComparatorPostprocessArgs[source]

基类: BaseArgs

比较器后处理:将后处理应用于输出。

parse_impl(args)[source]

解析命令行参数并填充以下属性

postprocess

将后处理函数名称映射到输出名称字典,输出名称映射到参数。例如,这可能类似于

{"top_k": {"output1": 5, "output2": 6}}
类型:

Dict[str, Dict[str, Any]]

add_to_script_impl(script, results_name)[source]
参数:

results_name (str) – 包含 Comparator.run() 结果的变量名称。

返回:

包含后处理结果的变量名称。这可能与原始名称相同。

返回类型:

str