(self)
| 157 | return self._consumer() |
| 158 | |
| 159 | def _consumer(self) -> Generator[Any, None, None]: |
| 160 | while True: |
| 161 | try: |
| 162 | yield self.get() |
| 163 | except StopIteration: |
| 164 | _logger.debug("Data consumer timed-out from get.") |
| 165 | return |
| 166 | |
| 167 | def _producer(self) -> None: |
| 168 | # pytorch dataloader is used here only because we need its sampler and multi-processing |