(
self,
cfg: TrainingConfig = None,
model: Model = None,
callbacks: Optional[Union[Callback, List[Callback]]] = None,
)
| 358 | |
| 359 | class TrainModel(SubModel): |
| 360 | def __init__( |
| 361 | self, |
| 362 | cfg: TrainingConfig = None, |
| 363 | model: Model = None, |
| 364 | callbacks: Optional[Union[Callback, List[Callback]]] = None, |
| 365 | ): |
| 366 | super().__init__("training", cfg, model, callbacks) |
| 367 | if not self._get_and_check_step(): |
| 368 | self.is_valid = False |
| 369 | if not self._get_and_check_opts(): |
| 370 | self.is_valid = False |
| 371 | if self.is_valid and (not self._get_and_check_jobs()): |
| 372 | self.is_valid = False |
| 373 | |
| 374 | def step(self, step_idx: int = 0): |
| 375 | assert self.is_valid, self.error_msg |
nothing calls this directly
no test coverage detected