默认评估函数 ¶

class torch.ao.quantization.default_eval_fn(model, calib_data)

定义默认评估函数。

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