Comparator
Module: polygraphy.tools.args
- class ComparatorRunArgs[source]
Bases:
BaseArgs
Comparator Inference: running inference via
Comparator.run()
.Depends on:
DataLoaderArgs
- parse_impl(args)[source]
Parses command-line arguments and populates the following attributes:
- warm_up
The number of warm-up runs to perform.
- Type:
int
- use_subprocess
Whether to run each runner in a subprocess.
- Type:
bool
- save_inputs_path
The path at which to save input data.
- Type:
str
- save_outputs_path
The path at which to save output data.
- Type:
str
- class ComparatorCompareArgs(allow_postprocessing: bool | None = None)[source]
Bases:
BaseArgs
Comparator Comparisons: inference output comparisons.
Depends on:
CompareFuncSimpleArgs
CompareFuncIndicesArgs
RunnerSelectArgs
ComparatorPostprocessArgs: if allow_postprocessing == True
- Parameters:
allow_postprocessing (bool) – Whether to post-processing of outputs before comparison. Defaults to True.
- parse_impl(args)[source]
Parses command-line arguments and populates the following attributes:
- validate
Whether to run output validation.
- Type:
bool
- load_outputs_paths
Path(s) from which to load outputs.
- Type:
List[str]
- fail_fast
Whether to fail fast.
- Type:
bool
- compare_func
The name of the comparison function to use.
- Type:
str
- compare_func_script
Path to a script defining a custom comparison function.
- Type:
str
- compare_func_name
The name of the function in the script that runs comparison.
- Type:
str