评价此页

default_eval_fn#

class torch.ao.quantization.default_eval_fn(model, calib_data)[source]#

定义默认的评估函数。

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