Method latest_checkpoint_path

axlearn/common/checkpointer.py:839–855

Returns the most recent (highest step count) complete checkpoint under base dir. Args: base_dir: Path to checkpoints dir. Returns: The path to the checkpoint directory under base_dir with the highest step count. The checkpoint is guaranteed to be

(cls, base_dir: str)

    @classmethod
    def latest_checkpoint_path(cls, base_dir: str) -> str:
        """Returns the most recent (highest step count) complete checkpoint under base dir.

        Args:
            base_dir: Path to checkpoints dir.

        Returns:
            The path to the checkpoint directory under base_dir with the highest step count.
            The checkpoint is guaranteed to be complete.
        """
        logging.log_first_n(
            logging.WARNING,
            msg="latest_checkpoint_path is deprecated. Use latest_checkpoint_step instead.",
            n=1,
        )
        # Note: checkpoint_paths should already filter incomplete checkpoints.
        return sorted(cls.checkpoint_paths(base_dir)).pop()

    @classmethod
    def checkpoint_steps(cls, base_dir: str) -> list[int]: