Class: Aws::SageMakerTrainingSessionRuntime::Types::Datum

Inherits:

Struct

Object
Struct
Aws::SageMakerTrainingSessionRuntime::Types::Datum

show all

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

Overview

A single training datum containing model input and optional loss function inputs.

Constant Summary collapse

SENSITIVE =

[:model_input]

Instance Attribute Summary collapse

#loss_fn_inputs ⇒ Hash<String,Types::TensorData>
The tensor inputs for the loss function, keyed by name.
#model_input ⇒ Types::ModelInput
The tokenized input to feed to the model.

Instance Attribute Details

#loss_fn_inputs ⇒ `Hash<String,Types::TensorData>`

The tensor inputs for the loss function, keyed by name. Unsupported keys are rejected. Supported keys: target_tokens, weights, advantages, logprobs, ref_log_prob, temperature. Required keys depend on the loss function type: SFT (cross_entropy) requires target_tokens and weights; RL (ppo, cispo, importance_sampling) requires target_tokens, weights, advantages, and logprobs.

Returns:

(Hash<String,Types::TensorData>)

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

class Datum < Struct.new(
  :model_input,
  :loss_fn_inputs)
  SENSITIVE = [:model_input]
  include Aws::Structure
end

#model_input ⇒ `Types::ModelInput`

The tokenized input to feed to the model.