Skip to content

Commit c9cf2bd

Browse files
committed
Dctor cache
1 parent ee7d8b4 commit c9cf2bd

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

paddle/fluid/operators/reader/create_double_buffer_reader_op.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ class DoubleBufferReader : public framework::DecoratedReader {
5252

5353
void ReadNextImpl(std::vector<framework::LoDTensor>* out) override;
5454

55+
~DoubleBufferReader() { Shutdown(); }
56+
5557
private:
5658
void ShutdownImpl() override {
5759
reader_->Shutdown();

python/paddle/fluid/tests/demo/pyreader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def main():
100100
trainer.run(fetch_list=[loss.name]))
101101
except fluid.core.EOFException:
102102
print 'End of epoch', epoch_id
103-
# train_reader.reset()
103+
train_reader.reset()
104104
train_data_thread.join()
105105

106106
test_data_thread = pipe_reader_to_queue(
@@ -111,7 +111,7 @@ def main():
111111
tester.run(fetch_list=[test_loss.name]))
112112
except fluid.core.EOFException:
113113
print 'End of testing'
114-
# test_reader.reset()
114+
test_reader.reset()
115115

116116
test_data_thread.join()
117117
break

0 commit comments

Comments
 (0)