refine patch

07d38a9b · fengjiayi · a571ef38 · 07d38a9b
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/operators/reader/create_shuffle_reader_op.cc paddle/fluid/operators/reader/create_shuffle_reader_op.cc +3 -0

未找到文件。
--- 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();