From e0edc080eba3d2293f73d6efcb714c6526a6d759 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=86=E6=96=87=E9=9C=9E?=
 <10511946+lu-wenxia@user.noreply.gitee.com>
Date: Sat, 7 May 2022 09:34:01 +0000
Subject: [PATCH] update

---
 TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py b/TensorFlow/contrib/cv/CycleGAN_ID0716_for_TensorFlow/reader.py
index fe6adb31c..04e82d2f4 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):
-- 
Gitee