Skip to content

LoadStateConfig

Source code in docs_src/capo_sagemakertrainingsessionruntime/types/load_state_config.py 
 6
 7
 8
 9
10
class LoadStateConfig(TypedDict, closed=True):
    model_package_arn: "str"
    """<p>The ARN of the model package containing the checkpoint to load. Must be a valid SageMaker model package ARN in the format <code>arn:aws:sagemaker:REGION:ACCOUNT:model-package/PACKAGE_NAME/VERSION</code>. Model package group ARNs are not accepted.</p>"""
    load_optimizer: "bool"
    """<p>Specifies whether to also restore the optimizer state alongside model weights. When true, optimizer state is restored to resume training from where it left off.</p>"""

load_optimizer instance-attribute 

load_optimizer: bool

Specifies whether to also restore the optimizer state alongside model weights. When true, optimizer state is restored to resume training from where it left off.

model_package_arn instance-attribute 

model_package_arn: str

The ARN of the model package containing the checkpoint to load. Must be a valid SageMaker model package ARN in the format arn:aws:sagemaker:REGION:ACCOUNT:model-package/PACKAGE_NAME/VERSION. Model package group ARNs are not accepted.