評價此頁

default_eval_fn#

class torch.ao.quantization.default_eval_fn(model, calib_data)[原始碼]#

定義預設的評估函式。

預設評估函式接受一個 `torch.utils.data.Dataset` 或一個輸入 Tensor 列表,並在資料集上執行模型。