fix datsset test=develop (#23043)

5 years ago · 660ff18488
parent 66fce9e824
commit 660ff18488
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/framework/data_set.cc
+++ b/paddle/fluid/framework/data_set.cc
@ -1087,6 +1087,7 @@ void MultiSlotDataset::SlotsShuffle(
  random_data.clear();
  random_data.shrink_to_fit();
  input_channel_->Close();
+  cur_channel_ = 0;

  timeline.Pause();
  VLOG(2) << "DatasetImpl<T>::SlotsShuffle() end"