Load model states from a checkpoint.
(self, dirpath: str)
| 571 | ) |
| 572 | |
| 573 | def _load_checkpoint(self, dirpath: str): |
| 574 | """Load model states from a checkpoint. |
| 575 | """ |
| 576 | stat_dict = flow.load(path=dirpath) |
| 577 | self._model.load_state_dict(stat_dict) |
| 578 | |
| 579 | def _save_checkpoint(self, dirpath: str): |
| 580 | """Save model states as a checkpoint. |
no test coverage detected