Skip to content

Commit a571ef3

Browse files
committed
fix bugs
1 parent c042137 commit a571ef3

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

paddle/fluid/operators/reader/create_double_buffer_reader_op.cc

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,11 +159,12 @@ void DoubleBufferReader::PrefetchThreadFunc() {
159159

160160
if (!buffer_->Send(&batch)) {
161161
VLOG(5) << "WARNING: The double buffer channel has been closed. The "
162-
"prefetch thread terminates.";
163-
break;
162+
"prefetch thread terminate.";
163+
return;
164164
}
165165
}
166166
buffer_->Close();
167+
VLOG(5) << "Prefetch thread terminates.";
167168
}
168169

169170
bool DoubleBufferReader::HasNext() const {

paddle/fluid/operators/reader/create_shuffle_reader_op.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,6 @@ class ShuffleReader : public framework::DecoratedReader {
5050
buffer_.clear();
5151
buffer_.reserve(buffer_size_);
5252
iteration_pos_ = 0;
53-
PADDLE_ENFORCE(reader_->HasNext());
5453
for (size_t i = 0; i < buffer_size_; ++i) {
5554
if (!reader_->HasNext()) {
5655
break;

0 commit comments

Comments
 (0)