diff --git a/TensorFlow/built-in/cv/image_classification/ShufflenetV1_ID0184_for_Tensorflow/input_pipeline.py b/TensorFlow/built-in/cv/image_classification/ShufflenetV1_ID0184_for_Tensorflow/input_pipeline.py
index ec5b026d2900fb1959cf6d44a86a481d85c5fb4a..6de748001f740cfa16f6449acbf2544af879b1c9 100644
--- a/TensorFlow/built-in/cv/image_classification/ShufflenetV1_ID0184_for_Tensorflow/input_pipeline.py
+++ b/TensorFlow/built-in/cv/image_classification/ShufflenetV1_ID0184_for_Tensorflow/input_pipeline.py
@@ -66,7 +66,7 @@ class Pipeline:
         # mix the training examples
         if is_training:
             dataset = dataset.shuffle(buffer_size=SHUFFLE_BUFFER_SIZE)
-        dataset = dataset.repeat(num_epochs)
+        dataset = dataset.repeat()
         
         # decode and augment data (对数据进行相关变换)
         dataset = dataset.apply(tf.contrib.data.map_and_batch(