(self)
| 151 | super().__init__() |
| 152 | |
| 153 | def run(self): |
| 154 | predictor = DefaultPredictor(self.cfg) |
| 155 | |
| 156 | while True: |
| 157 | task = self.task_queue.get() |
| 158 | if isinstance(task, AsyncPredictor._StopToken): |
| 159 | break |
| 160 | idx, data = task |
| 161 | result = predictor(data) |
| 162 | self.result_queue.put((idx, result)) |
| 163 | |
| 164 | def __init__(self, cfg, num_gpus: int = 1): |
| 165 | """ |
no test coverage detected