(self)
| 410 | self._call_model_initializer() |
| 411 | |
| 412 | def _call_model_initializer(self): |
| 413 | model_weight_initializer = instantiate( |
| 414 | self.checkpoint_conf.model_weight_initializer |
| 415 | ) |
| 416 | if model_weight_initializer is not None: |
| 417 | logging.info( |
| 418 | f"Loading pretrained checkpoint from {self.checkpoint_conf.model_weight_initializer}" |
| 419 | ) |
| 420 | self.model = model_weight_initializer(model=self.model) |
| 421 | |
| 422 | def _load_resuming_checkpoint(self, ckpt_path: str): |
| 423 | logging.info(f"Resuming training from {ckpt_path}") |
no test coverage detected