LoadStateConfig
The configuration for loading training state from a previously saved checkpoint. This restores model weights and optionally optimizer state.
Contents
- loadOptimizer
-
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.
Type: Boolean
Required: Yes
- modelPackageArn
-
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.Type: String
Required: Yes
See Also
For more information about using this API in one of the language-specific Amazon SDKs, see the following: