Method checkpoint_steps

axlearn/common/checkpointer.py:1019–1020

(cls, base_dir: str)

    @classmethod
    def checkpoint_steps(cls, base_dir: str) -> list[int]:
        return [parse_step_from_dir(path) for path in cls.checkpoint_paths(base_dir)]

    @classmethod
    def cleanup_checkpoint(cls, ckpt_dir: str, *, sync: bool = True):