Obtains the checkpoint dir for the given step.
(self, step: int)
| 1101 | logging.info("GC loop done") |
| 1102 | |
| 1103 | def ckpt_dir(self, step: int) -> str: |
| 1104 | """Obtains the checkpoint dir for the given step.""" |
| 1105 | cfg: Checkpointer.Config = self.config |
| 1106 | return build_step_dir(cfg.dir, step=step) |
| 1107 | |
| 1108 | def save( |
| 1109 | self, *, step: int, state: NestedTensor, evaler_summaries: Optional[dict[str, Any]] = None |