Method update

ppq/quantization/analyse/util/__init__.py:32–47

(self, y_pred: torch.Tensor, y_real: torch.Tensor)

        self.reduce     = reduce

    def update(self, y_pred: torch.Tensor, y_real: torch.Tensor):
        elements = y_pred.shape[0]
        if elements != y_real.shape[0]:
            raise Exception(
                'Can not update measurement, cause your input data do not share a same batchsize. '
                f'Shape of y_pred {y_pred.shape} - against shape of y_real {y_real.shape}')
        result = self.measure_fn(y_pred=y_pred, y_real=y_real).item()

        if self.reduce == 'mean':
            self.measure = self.measure * self.num_of_elements + result * elements
            self.num_of_elements += elements
            self.measure /= self.num_of_elements

        if self.reduce == 'max':
            self.measure = max(self.measure, result)
            self.num_of_elements += elements

class MeasurePrinter():
    """Helper class for print top-k record."""