(self, loader)
| 19 | """ |
| 20 | |
| 21 | def __init__(self, loader): |
| 22 | self.loader = iter(loader) |
| 23 | self.stream = torch.cuda.Stream() |
| 24 | self.input_cuda = self._input_cuda_for_image |
| 25 | self.record_stream = DataPrefetcher._record_stream_for_image |
| 26 | self.preload() |
| 27 | |
| 28 | def preload(self): |
| 29 | try: |