(self)
| 77 | self._world_size = world_size |
| 78 | |
| 79 | def __iter__(self): |
| 80 | start = self._rank |
| 81 | yield from itertools.islice( |
| 82 | self._infinite_indices(), start, None, self._world_size |
| 83 | ) |
| 84 | |
| 85 | def _infinite_indices(self): |
| 86 | g = torch.Generator() |
no test coverage detected