提交 589467f2 编写于 作者: X xjqbest 提交者: dongdaxiang

fix bug

上级 b7940c29
...@@ -221,9 +221,9 @@ void DatasetImpl<T>::DestroyReaders() { ...@@ -221,9 +221,9 @@ void DatasetImpl<T>::DestroyReaders() {
} }
std::vector<std::shared_ptr<paddle::framework::DataFeed>>().swap(readers_); std::vector<std::shared_ptr<paddle::framework::DataFeed>>().swap(readers_);
VLOG(3) << "readers size: " << readers_.size(); VLOG(3) << "readers size: " << readers_.size();
// if memory_data_ is not empty, which means it's not InMemory mode, // if memory_data_ is empty, which means it's not InMemory mode,
// so the next epoch should read all data again // so the next epoch should read all data again
if (memory_data_.size() != 0) { if (memory_data_.size() == 0) {
file_idx_ = 0; file_idx_ = 0;
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册