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

fix bug

上级 b7940c29
......@@ -221,9 +221,9 @@ void DatasetImpl<T>::DestroyReaders() {
}
std::vector<std::shared_ptr<paddle::framework::DataFeed>>().swap(readers_);
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
if (memory_data_.size() != 0) {
if (memory_data_.size() == 0) {
file_idx_ = 0;
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册