(self)
| 151 | self._gpus = gpus |
| 152 | |
| 153 | def _setup_graph(self): |
| 154 | # Create predictors on the available predictor GPUs. |
| 155 | num_gpu = len(self._gpus) |
| 156 | predictors = [self.trainer.get_predictor( |
| 157 | ['state'], ['policy', 'pred_value'], |
| 158 | self._gpus[k % num_gpu]) |
| 159 | for k in range(PREDICTOR_THREAD)] |
| 160 | self.async_predictor = MultiThreadAsyncPredictor( |
| 161 | predictors, batch_size=PREDICT_BATCH_SIZE) |
| 162 | |
| 163 | def _before_train(self): |
| 164 | self.async_predictor.start() |
nothing calls this directly
no test coverage detected