MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / step

Method step

python/oneflow/framework/model.py:495–510  ·  view source on GitHub ↗
(self, step_idx: int = 0)

Source from the content-addressed store, hash-verified

493 self.is_valid = False
494
495 def step(self, step_idx: int = 0):
496 assert self.is_valid
497 if (step_idx + 1) % self._cfg.step_interval == 0:
498 outputs = None
499 if self._is_numpy_input:
500 batch = None
501 if step_idx == 0:
502 batch = self._first_numpy_batch
503 else:
504 batch = self._cfg.data(step_idx, 0)
505 outputs = self._job(*batch).get()
506 else:
507 outputs = self._job().get()
508 self._method_callback(
509 "on_validation_step_end", step_idx=step_idx, outputs=outputs
510 )
511
512 def _get_and_check_step(self):
513 if not self._model.method_overrided("validation_step"):

Callers

nothing calls this directly

Calls 3

_method_callbackMethod · 0.80
dataMethod · 0.45
getMethod · 0.45

Tested by

no test coverage detected