Method
__init__
(self, generator, local_rank, max_prefetch=6)
Source from the content-addressed store, hash-verified
| 78 | |
| 79 | class BackgroundGenerator(threading.Thread): |
| 80 | def __init__(self, generator, local_rank, max_prefetch=6): |
| 81 | super(BackgroundGenerator, self).__init__() |
| 82 | self.queue = Queue.Queue(max_prefetch) |
| 83 | self.generator = generator |
| 84 | self.local_rank = local_rank |
| 85 | self.daemon = True |
| 86 | self.start() |
| 87 | |
| 88 | def run(self): |
| 89 | torch.cuda.set_device(self.local_rank) |
Callers
nothing calls this directly
Tested by
no test coverage detected