revert-16555-model_data_cryption_link_all_lib
xjqbest 6 years ago committed by dongdaxiang
parent b7940c2918
commit 589467f24c

@ -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;
} }
} }

Loading…
Cancel
Save