Shortcuts

default_eval_fn

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

定义默认的评估函数。

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