()
| 438 | |
| 439 | def _construct_job(self, optimizer_idx: int = 0): |
| 440 | def job(): |
| 441 | batch = self._cfg.data(0, optimizer_idx) |
| 442 | outputs = self._model.training_step( |
| 443 | batch=batch, optimizer_idx=optimizer_idx |
| 444 | ) |
| 445 | loss = None |
| 446 | if isinstance(outputs, tuple) and len(outputs) > 0: |
| 447 | loss = outputs[0] |
| 448 | else: |
| 449 | loss = outputs |
| 450 | self._opts[optimizer_idx].minimize(loss) |
| 451 | return outputs |
| 452 | |
| 453 | job.__name__ = ( |
| 454 | self._model.__class__.__name__ + "_Model_train_job_" + str(optimizer_idx) |
nothing calls this directly
no test coverage detected