def METHOD_NAME(self, input: Dict[str,
                              Tensor]) -> Dict[str, Union[list, Tensor]]:
    """return the result of the model
    Args:
        inputs (Tensor): the preprocessed data
    Returns:
        Dict[str, Tensor]: results
    """
    if self.training:
        return self._train_forward(**input)
    elif 'target' in input:
        return self._evaluate_postprocess(**input)
    else:
        return self.model(**input)