def METHOD_NAME(self):
    batch = []
    for i, file in enumerate(os.listdir(self.inputdir)):
        if self.subset_list is not None and i not in self.subset_list:
            continue
        batch.append(self._vqa_file_split(file))
        if len(batch) == self.batch:
            temp = batch
            batch = []
            yield list(map(list, zip(*temp)))
    for i in range(1):
        yield list(map(list, zip(*batch)))