Class: Aws::SageMakerTrainingSessionRuntime::Types::OperationConfig

Inherits:
Struct
  • Object
show all
Defined in:
gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb

Overview

Note:

OperationConfig is a union - when making an API calls you must set exactly one of the members.

The configuration for a training operation. Exactly one member must be set to specify the operation type and its parameters.

See Also:

Defined Under Namespace

Classes: Forward, ForwardBackward, LoadState, OptimizerStep, SaveState, SaveWeights, Unknown

Constant Summary collapse

SENSITIVE = 
[]

Instance Attribute Summary collapse

Instance Attribute Details

#forward ⇒ Types::ForwardConfig

The configuration for a forward-only pass through the model without gradient computation.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#forward_backward ⇒ Types::ForwardBackwardConfig

The configuration for a forward and backward pass that computes gradients.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#load_state ⇒ Types::LoadStateConfig

The configuration for loading training state from a previously saved checkpoint.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#optimizer_step ⇒ Types::OptimizerStepConfig

The configuration for an optimizer step that updates model weights using accumulated gradients.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#save_state ⇒ Types::SaveStateConfig

The configuration for saving full training state, including model weights and optimizer state.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#save_weights ⇒ Types::SaveWeightsConfig

The configuration for saving model weights as a checkpoint.

Returns:



578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

class OperationConfig < Struct.new(
  :forward,
  :forward_backward,
  :optimizer_step,
  :save_weights,
  :save_state,
  :load_state,
  :unknown)
  SENSITIVE = []
  include Aws::Structure
  include Aws::Structure::Union

  class Forward < OperationConfig; end
  class ForwardBackward < OperationConfig; end
  class OptimizerStep < OperationConfig; end
  class SaveWeights < OperationConfig; end
  class SaveState < OperationConfig; end
  class LoadState < OperationConfig; end
  class Unknown < OperationConfig; end
end

#unknown ⇒ Object

Returns the value of attribute unknown

Returns:

  • (Object) —

    the current value of unknown



578
579
580
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 578

def unknown
  @unknown
end