From a571ef382ee45b1e255f44c9650510f20403f4bf Mon Sep 17 00:00:00 2001
From: fengjiayi <fengjiayi@baidu.com>
Date: Mon, 19 Mar 2018 14:11:40 +0800
Subject: [PATCH] fix bugs
---
.../fluid/operators/reader/create_double_buffer_reader_op.cc | 5 +++--
paddle/fluid/operators/reader/create_shuffle_reader_op.cc | 1 -
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc b/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc
index d0de092947..8960fe5d63 100644
--- a/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc
+++ b/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc
@@ -159,11 +159,12 @@ void DoubleBufferReader::PrefetchThreadFunc() {
if (!buffer_->Send(&batch)) {
VLOG(5) << "WARNING: The double buffer channel has been closed. The "
- "prefetch thread terminates.";
- break;
+ "prefetch thread terminate.";
+ return;
}
}
buffer_->Close();
+ VLOG(5) << "Prefetch thread terminates.";
}
bool DoubleBufferReader::HasNext() const {
diff --git a/paddle/fluid/operators/reader/create_shuffle_reader_op.cc b/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
index 70e2f587dc..4ebef4aed7 100644
--- a/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
+++ b/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
@@ -50,7 +50,6 @@ class ShuffleReader : public framework::DecoratedReader {
buffer_.clear();
buffer_.reserve(buffer_size_);
iteration_pos_ = 0;
- PADDLE_ENFORCE(reader_->HasNext());
for (size_t i = 0; i < buffer_size_; ++i) {
if (!reader_->HasNext()) {
break;