| 99 | } |
| 100 | |
| 101 | void DataLoader::clean() { |
| 102 | if (mJobs != nullptr) { |
| 103 | join(); |
| 104 | mWorkers.clear(); |
| 105 | mJobs->clear(); |
| 106 | mDataQueue->clear(); |
| 107 | } |
| 108 | // should reset sampler before prefetch |
| 109 | mSampler->reset(mSampler->size()); |
| 110 | } |
| 111 | size_t DataLoader::size() const { |
| 112 | return mDataset->size(); |
| 113 | } |