Interface LoadStateConfig.Builder

All Superinterfaces:

Buildable, CopyableBuilder<LoadStateConfig.Builder,LoadStateConfig>, SdkBuilder<LoadStateConfig.Builder,LoadStateConfig>, SdkPojo

Enclosing class:

LoadStateConfig

@Mutable
@NotThreadSafe
public static interface LoadStateConfig.Builder
extends SdkPojo, CopyableBuilder<LoadStateConfig.Builder,LoadStateConfig>

Method Summary

Modifier and Type	Method	Description
LoadStateConfig.Builder	loadOptimizer(Boolean loadOptimizer)	Specifies whether to also restore the optimizer state alongside model weights.
LoadStateConfig.Builder	modelPackageArn(String modelPackageArn)	The ARN of the model package containing the checkpoint to load.

Methods inherited from interface software.amazon.awssdk.utils.builder.CopyableBuilder

copy

Methods inherited from interface software.amazon.awssdk.utils.builder.SdkBuilder

applyMutation, build

Methods inherited from interface software.amazon.awssdk.core.SdkPojo

equalsBySdkFields, sdkFieldNameToField, sdkFields

Method Details

modelPackageArn

LoadStateConfig.Builder modelPackageArn(String 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.

Parameters:

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.

Returns:

Returns a reference to this object so that method calls can be chained together.

loadOptimizer

LoadStateConfig.Builder loadOptimizer(Boolean 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.

Parameters:

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.

Returns:

Returns a reference to this object so that method calls can be chained together.