| 112 | return mDataset->size(); |
| 113 | } |
| 114 | size_t DataLoader::iterNumber() const { |
| 115 | auto number = mDataset->size(); |
| 116 | auto batch = mConfig->batchSize; |
| 117 | auto dropLast = mConfig->dropLast; |
| 118 | if (dropLast) { |
| 119 | return number / batch; |
| 120 | } |
| 121 | return ((int)number + (int)batch - 1) / (int)batch; |
| 122 | } |
| 123 | |
| 124 | |
| 125 | DataLoader* DataLoader::makeDataLoader(std::shared_ptr<BatchDataset> dataset, |