fixed evaluate function interface error
@@ -1352,7 +1352,8 @@ class PyTorch_IPEXAdaptor(TemplateAdaptor): # pragma: no cover
json.dump(self.cfgs, write_f)
def evaluate(self, model, dataloader, postprocess=None,
- metric=None, measurer=None, iteration=-1, tensorboard=False):
+ metric=None, measurer=None, iteration=-1,
+ tensorboard=False, fp32_baseline=False):
"""Execute the evaluate process on the specified model.
Args:
@@ -1364,6 +1365,7 @@ class PyTorch_IPEXAdaptor(TemplateAdaptor): # pragma: no cover
iteration (int, optional): number of iterations to evaluate.
tensorboard (bool, optional): dump output tensor to tensorboard summary
files(IPEX unspport).
+ fp32_baseline (boolen, optional): only for compare_label=False pipeline
Returns:
(dict): quantized model
