diff --git a/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py b/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py index fe6adb31cf43b35cbfa567334aa6648f520bb629..04e82d2f4dbd27b04cc780724666d8db444d637a 100644 --- a/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py +++ b/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py @@ -56,7 +56,7 @@ class Reader(): dataset = tf.data.TFRecordDataset(self.tfrecords_file, buffer_size=256 << 20) data_ = dataset.map(self.parse_function, num_parallel_calls=tf.data.experimental.AUTOTUNE).shuffle( self.min_queue_examples + 3 * self.batch_size).repeat().batch(self.batch_size, drop_remainder=True) - data = data_.make_one_shot_iterator() + data = data_.make_initializable_iterator() return data.get_next() def parse_function(self, example_proto):