Method latest_checkpoint_step

axlearn/common/checkpointer.py:870–873

Returns the most recent (highest step count) checkpoint step under base dir.

(cls, base_dir: str)

    @classmethod
    def latest_checkpoint_step(cls, base_dir: str) -> int:
        """Returns the most recent (highest step count) checkpoint step under base dir."""
        # Note: checkpoint_steps should already filter incomplete checkpoints.
        return max(cls.checkpoint_steps(base_dir))

    def __init__(self, cfg: Module.Config, *, parent: Optional[Module]):
        super().__init__(cfg, parent=parent)