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):