Method __init__

python/oneflow/framework/model.py:587–597

(
        self,
        cfg: TrainingConfig = None,
        model: Model = None,
        callbacks: Optional[Union[Callback, List[Callback]]] = None,
    )

class TrainModelOOPStyle(SubModel):
    def __init__(
        self,
        cfg: TrainingConfig = None,
        model: Model = None,
        callbacks: Optional[Union[Callback, List[Callback]]] = None,
    ):
        super().__init__("training", cfg, model, callbacks)
        if not self._get_and_check_step():
            self.is_valid = False
        if not self._get_and_check_opts():
            self.is_valid = False

    def step(self, step_idx: int = 0):
        assert self.is_valid, self.error_msg