Force to wait for the async checkpoint saving (if any) to finish.
(self)
| 574 | f"`Controller.__init__` before calling `{for_method}()`.") |
| 575 | |
| 576 | def _sync_on_async_checkpointing(self): |
| 577 | """Force to wait for the async checkpoint saving (if any) to finish.""" |
| 578 | # pylint: disable=protected-access |
| 579 | if self.checkpoint_manager: |
| 580 | logging.info("Sync on async checkpoint saving.") |
| 581 | self.checkpoint_manager.sync() |
| 582 | |
| 583 | |
| 584 | class StepTimer: |