Return the number of batches.
(self)
| 222 | yield batch |
| 223 | |
| 224 | def __len__(self): |
| 225 | """Return the number of batches.""" |
| 226 | if self.drop_last: |
| 227 | return len(self.sampler) // self.batch_size |
| 228 | else: |
| 229 | return (len(self.sampler) + self.batch_size - 1) // self.batch_size |
| 230 | |
| 231 | |
| 232 | class BaseDataLoader: |