(self, step_idx: int = 0)
| 659 | self.is_valid = False |
| 660 | |
| 661 | def step(self, step_idx: int = 0): |
| 662 | assert self.is_valid |
| 663 | if (step_idx + 1) % self._cfg.step_interval == 0: |
| 664 | outputs = None |
| 665 | with oneflow._oneflow_internal.autograd.no_grad(): |
| 666 | inputs = self._cfg.data(step_idx, 0) |
| 667 | model_previous_mode = self._model.training |
| 668 | self._model.train() |
| 669 | outputs = self._model.validation_step(inputs) |
| 670 | self._model.train(model_previous_mode) |
| 671 | self._method_callback( |
| 672 | "on_validation_step_end", step_idx=step_idx, outputs=outputs |
| 673 | ) |
| 674 | |
| 675 | def _get_and_check_step(self): |
| 676 | if not self._model.method_overrided("validation_step"): |
nothing calls this directly
no test coverage detected