(self)
| 337 | self.start_index = 0 |
| 338 | |
| 339 | def __iter__(self): |
| 340 | num_buffers = self.dataset.num_buffers |
| 341 | len_buffer = self.dataset.len_buffer |
| 342 | num_buffers_i = num_buffers // self.num_replicas |
| 343 | num_samples_i = len_buffer * num_buffers_i |
| 344 | |
| 345 | indices_i = ( |
| 346 | np.arange(self.start_index, num_samples_i) + |
| 347 | self.rank * num_samples_i) |
| 348 | indices_i = indices_i.tolist() |
| 349 | |
| 350 | return iter(indices_i) |
| 351 | |
| 352 | def reset(self): |
| 353 | self.start_index = 0 |