评价此页

default_eval_fn#

class torch.ao.quantization.default_eval_fn(model, calib_data)[源代码]#

定义默认的评估函数。

默认评估函数接受一个 `torch.utils.data.Dataset` 或一个输入 Tensor 列表,并在数据集上运行模型。