default_eval_fn

classtorch.ao.quantization.default_eval_fn(model, calib_data)[源代码]

定义默认的评价函数。

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

本页目录