sagemaker.core.remote_function.checkpoint_location

This module is used to define the CheckpointLocation to remote function.

Classes

CheckpointLocation(s3_uri)

Class to represent the location where checkpoints are accessed in a remote function.

class sagemaker.core.remote_function.checkpoint_location.CheckpointLocation(s3_uri)

Bases: PathLike

Class to represent the location where checkpoints are accessed in a remote function.

To save or load checkpoints in a remote function, pass an CheckpointLocation object as a function parameter and use it as a os.PathLike object. This CheckpointLocation object represents the local directory (/opt/ml/checkpoints/) of checkpoints in side the job.