Sets random seed prior to thread initialization

apache · Sep 11, 2018 · 0deffe9 · 0deffe9
1 parent 6aa8d75
commit 0deffe9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/io/iter_image_recordio.cc b/src/io/iter_image_recordio.cc
@@ -264,6 +264,8 @@ class ImageRecordIter : public IIterator<DataInst> {
     parser_.Init(kwargs);
     // prefetch at most 4 minbatches
     iter_.set_max_capacity(4);
+    // set random seed
+    rnd_.seed(kRandMagic + param_.seed);
     // init thread iter
     iter_.Init([this](std::vector<InstVector<DType>> **dptr) {
         if (*dptr == nullptr) {
@@ -273,7 +275,6 @@ class ImageRecordIter : public IIterator<DataInst> {
       },
       [this]() { parser_.BeforeFirst(); });
     inst_ptr_ = 0;
-    rnd_.seed(kRandMagic + param_.seed);
   }
   // before first
   virtual void BeforeFirst(void) {