refine patch

7 years ago · 07d38a9b9a
parent a571ef382e
commit 07d38a9b9a
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
+++ b/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
@ -34,6 +34,9 @@ class ShuffleReader : public framework::DecoratedReader {
  }

  void ReadNext(std::vector<framework::LoDTensor>* out) override {
+    if (!HasNext()) {
+      PADDLE_THROW("There is no next data!");
+    }
    if (iteration_pos_ >= buffer_.size()) {
      VLOG(10) << "Resetting shuffle buffer";
      ReadIntoBuffers();