]> Gitweb @ Texas Instruments - Open Source Git Repositories - git.TI.com/gitweb - jacinto-ai/caffe-jacinto.git/commitdiff
Merge branch 'nvcaffe-17.05.1' into nvcaffe-17.05.1shuffle
authorSergei Nikolaev <snikolaev@nvidia.com>
Sun, 7 May 2017 08:23:07 +0000 (01:23 -0700)
committerSergei Nikolaev <snikolaev@nvidia.com>
Sun, 7 May 2017 08:23:07 +0000 (01:23 -0700)
1  2 
src/caffe/data_reader.cpp
src/caffe/layers/data_layer.cpp

index db9e407f88f72f3a744e01afa4b26504b4ee1111,7c180167e47420779187634a6b615b421e621007..80f8b6c3b1389bac86b2976aa10c8f116a081867
@@@ -163,16 -122,12 +163,17 @@@ void DataReader::CursorManager::next(Da
      rec_id_ += full_cycle_ - batch_size_;
      rec_end_ += full_cycle_;
    }
-   for (size_t i = old_id; i < rec_id_; ++i) {
+   size_t rnd_skip = 0UL;  // batch_size_ > 100 ? (caffe_rng_rand() % 5) : 0UL;
+   for (size_t i = old_id; i < rec_id_ + rnd_skip; ++i) {
      cursor_->Next();
      if (!cursor_->valid()) {
 -      LOG_IF(INFO, solver_rank_ == 0 && parser_thread_id_ == 0)
 -          << "Restarting data pre-fetching";
 +      if(cache_) {
 +        LOG(INFO) << "Solver " << solver_rank_ << ", parser " << parser_thread_id_
 +            << ", records cached ";
 +
 +        break;  // we cache first epoch, then we just read it from cache
 +      }
 +      LOG_IF(INFO, solver_rank_ == 0 && parser_thread_id_ == 0) << "Restarting data pre-fetching";
        cursor_->SeekToFirst();
      }
    }
Simple merge