From 589467f24c35f2b4955b8ef6a795c38a6f439e8c Mon Sep 17 00:00:00 2001
From: xjqbest <173596896@qq.com>
Date: Fri, 22 Mar 2019 13:12:59 +0800
Subject: [PATCH] fix bug

---
 paddle/fluid/framework/data_set.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/framework/data_set.cc b/paddle/fluid/framework/data_set.cc
index e9bf392d69..c2e8bff348 100644
--- a/paddle/fluid/framework/data_set.cc
+++ b/paddle/fluid/framework/data_set.cc
@@ -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;
   }
 }
-- 
GitLab