Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 3526ac11 编写于 作者: F fengjiayi
浏览文件
操作

refine code

上级 744cc412
显示空白变更内容
内联 并排
Showing with 1 addition and 3 deletion
paddle/fluid/operators/reader/open_files_op.cc
浏览文件 @ 3526ac11
...@@ -63,9 +63,6 @@ void MultiFileReader::ReadNext(std::vector<framework::LoDTensor>* out) { ...@@ -63,9 +63,6 @@ void MultiFileReader::ReadNext(std::vector<framework::LoDTensor>* out) {
void MultiFileReader::ReInit() { void MultiFileReader::ReInit() {
EndScheduler(); EndScheduler();
for (auto& reader : readers_) {
reader->ReInit();
}
StartNewScheduler(); StartNewScheduler();
} }
...@@ -141,6 +138,7 @@ void MultiFileReader::PrefetchThreadFunc(size_t reader_idx, size_t thread_idx) { ...@@ -141,6 +138,7 @@ void MultiFileReader::PrefetchThreadFunc(size_t reader_idx, size_t thread_idx) {
std::vector<framework::LoDTensor> ins; std::vector<framework::LoDTensor> ins;
reader->ReadNext(&ins); reader->ReadNext(&ins);
if (ins.empty()) { if (ins.empty()) {
reader->ReInit();
break; break;
} }
try { try {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或