Method restore_from_dir

axlearn/common/checkpointer.py:334–342

(
        self,
        step: int,
        state: Union[NestedTensor, NestedTensorSpec],
        *,
        ckpt_dir: str,
        validation: CheckpointValidationType = CheckpointValidationType.EXACT,
    )

        raise NotImplementedError(type(self))

    def restore_from_dir(
        self,
        step: int,
        state: Union[NestedTensor, NestedTensorSpec],
        *,
        ckpt_dir: str,
        validation: CheckpointValidationType = CheckpointValidationType.EXACT,
    ) -> NestedTensor:
        raise NotImplementedError(type(self))

    def stop(self):
        """Stops and disposes resources."""