比较函数
模块: polygraphy.tools.args
- class CompareFuncSimpleArgs[source]
基类:
BaseArgs比较函数: simple: CompareFunc.simple 比较函数。
依赖于
ComparatorCompareArgs
- parse_impl(args)[source]
解析命令行参数并填充以下属性
- no_shape_check
是否跳过形状检查。
- 类型:
bool
- rtol
每张量相对容差。
- 类型:
Dict[str, float]
- atol
每张量绝对容差。
- 类型:
Dict[str, float]
- check_error_stat
要检查的误差指标。
- 类型:
str
- infinities_compare_equal
是否允许将正负无穷大视为相等进行比较。
- 类型:
bool
- save_heatmaps
保存误差热图的目录。
- 类型:
str
- show_heatmaps
是否显示误差热图。
- 类型:
bool
- save_error_metrics_plot
存储生成的误差图表的路径。
- 类型:
str
- show_error_metrics_plot
是否显示误差指标图表。
- 类型:
bool
- error_quantile
要计算的每张量误差分位数。
- 类型:
Dict[str, float]