(self)
| 497 | self.batch_storage.push([inputs_batch, targets_batch]) |
| 498 | |
| 499 | def run(self): |
| 500 | start = False |
| 501 | while True: |
| 502 | # waiting for start signal |
| 503 | if not start: |
| 504 | start = ray.get(self.storage.get_start_signal.remote()) |
| 505 | time.sleep(0.1) |
| 506 | continue |
| 507 | |
| 508 | trained_steps = ray.get(self.storage.get_counter.remote()) |
| 509 | if trained_steps >= self.config.training_steps + self.config.last_steps: |
| 510 | time.sleep(30) |
| 511 | break |
| 512 | |
| 513 | self._prepare_target_gpu() |
nothing calls this directly
no test coverage detected