def METHOD_NAME(self, losses, detections):
    # type: (Dict[str, Tensor], List[Dict[str, Tensor]]) -> Union[Dict[str, Tensor], List[Dict[str, Tensor]]]
    if self.training:
        return losses
    return detections