def METHOD_NAME(self) -> Tuple[torch.Tensor]:
    self.model.METHOD_NAME()
    with torch.no_grad():
        for _batch_id, (images, _targets) in zip(range(self.NUM_OF_BATCHES), self.data_loader):
            out = self.model(images)
    out = list(map(lambda x: x.values(), out))
    return tuple(itertools.chain(*out))