def METHOD_NAME(self):
    """ Recommended
    Returns model, example_inputs
    Both model and example_inputs should be on self.device properly.
    `model(*example_inputs)` should execute one step of model forward.
    """
    images = []
    for (i, _) in self.example_inputs:
        images = (i[0], i[1])
    return self.model, images