Class: Aws::SageMakerTrainingSessionRuntime::Types::LoadStateConfig

Inherits:

Struct

Object
Struct
Aws::SageMakerTrainingSessionRuntime::Types::LoadStateConfig

show all

Defined in:: gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb

Overview

The configuration for loading training state from a previously saved checkpoint. This restores model weights and optionally optimizer state.

Constant Summary collapse

SENSITIVE =

[]

Instance Attribute Summary collapse

#load_optimizer ⇒ Boolean
Specifies whether to also restore the optimizer state alongside model weights.
#model_package_arn ⇒ String
The ARN of the model package containing the checkpoint to load.

Instance Attribute Details

#load_optimizer ⇒ `Boolean`

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:

(Boolean)

# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 488

class LoadStateConfig < Struct.new(
  :model_package_arn,
  :load_optimizer)
  SENSITIVE = []
  include Aws::Structure
end

#model_package_arn ⇒ `String`

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.