022034282fc8ca8694ceffc2aec711c9376a3e37,hypergan/inputs/image_loader.py,ImageLoader,tfrecord_create,#ImageLoader#Any#Any#Any#Any#Any#Any#Any#,18

Before Change

        dataset = dataset.map(parse_function, num_parallel_calls=4)
        dataset = dataset.flat_map(lambda x: x.batch(self.batch_size, drop_remainder=True))
        dataset = dataset.repeat()
        dataset = dataset.prefetch(10)
        dataset = dataset.map(set_shape)

        self.dataset = dataset
        return dataset

After Change

            dataset = dataset.shuffle(self.file_count)
        dataset = dataset.map(parse_function, num_parallel_calls=4)
        dataset = dataset.flat_map(lambda x: x.batch(self.batch_size, drop_remainder=True).repeat().prefetch(10))
        dataset = dataset.repeat().prefetch(10)
        dataset = dataset.map(set_shape)

        self.dataset = dataset
        return dataset

Instances