--evaluator

功能说明

基于"Evaluator"基类并且包含evaluator评估器的python脚本。

如果用户没有获取到模型匹配的bin格式数据集，或者用户想使用自己的脚本进行训练后量化，则需要使用该参数。

关联参数

该参数与--input_shape、--data_dir、--data_types不能同时使用。

参数取值

参数值：包含evaluator评估器的python脚本。

参数值约束：无

推荐配置及收益

无。

示例

--evaluator=EVALUATOR.py

脚本样例如下，用户需要使用原始待量化的模型和测试集，实现回调函数calibration()。其中：calibration()完成校准的推理。

1
2
3
4
5

class ModelEvaluator(AutoCalibrationEvaluatorBase):
      # The evaluator for model

    def calibration(self, graph, outputs):
        # 进行模型的校准推理，推理的batch数要和量化配置的batch_num一致

使用约束

该参数只在训练后量化场景使用。