def get_data(self, data_state, request=None):

        if request is None:
            request = [self.current_index]
            self.current_index += 1

        return self.load_images(request)
Get a new sample of data
dataset/ETHPy150Open Sohl-Dickstein/Diffusion-Probabilistic-Models/imagenet_data.py/IMAGENET.get_data