| 68 | |
| 69 | template<typename Dtype> |
| 70 | void DataLayer<Dtype>::Next() { |
| 71 | cursor_->Next(); |
| 72 | if (!cursor_->valid()) { |
| 73 | LOG_IF(INFO, Caffe::root_solver()) |
| 74 | << "Restarting data prefetching from start."; |
| 75 | cursor_->SeekToFirst(); |
| 76 | } |
| 77 | offset_++; |
| 78 | } |
| 79 | |
| 80 | // This function is called on prefetch thread |
| 81 | template<typename Dtype> |