Comparator

模块: polygraphy.tools.args

class ComparatorRunArgs

基类: BaseArgs

Comparator 推理: 通过 Comparator.run() 运行推理。

依赖于

• DataLoaderArgs

parse_impl(args)

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

warm_up

要执行的预热运行次数。

类型:

int

use_subprocess

是否在子进程中运行每个 runner。

类型:

bool

save_inputs_path

保存输入数据的路径。

类型:

str

save_outputs_path

保存输出数据的路径。

类型:

str

class ComparatorCompareArgs(allow_postprocessing: bool | None = None)

基类: BaseArgs

Comparator 比较: 推理输出比较。

依赖于

• CompareFuncSimpleArgs

• CompareFuncIndicesArgs

• RunnerSelectArgs

• ComparatorPostprocessArgs: 如果 allow_postprocessing == True

参数:

allow_postprocessing (bool) – 是否在比较前对输出进行后处理。默认为 True。

parse_impl(args)

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

validate

是否运行输出验证。

类型:

bool

load_outputs_paths

从中加载输出的路径。

类型:

List[str]

fail_fast

是否快速失败。

类型:

bool

compare_func

要使用的比较函数的名称。

类型:

str

compare_func_script

定义自定义比较函数的脚本路径。

类型:

str

compare_func_name

脚本中运行比较的函数的名称。

类型:

str

add_to_script_impl(script, results_name)

参数:

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

返回:

包含 Comparator.compare_accuracy() 状态的变量名称。

返回类型:

str