def METHOD_NAME(self, num_batches:int=1) -> Tuple[Generator, Optional[int]]:
    def _gen_inputs():
        while True:
            result = []
            for _i in range(num_batches):
                result.append((torch.randn((self.batch_size, 3, 224, 224)).to(self.device),))
            if self.dargs.precision == "fp16":
                result = list(map(lambda x: (x[0].half(), ), result))
            yield result
    return (_gen_inputs(), None)