Datum
A single training datum containing model input and optional loss function inputs.
Contents
- modelInput
-
The tokenized input to feed to the model.
Type: ModelInput object
Required: Yes
- lossFnInputs
-
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) requirestarget_tokensandweights; RL (ppo,cispo,importance_sampling) requirestarget_tokens,weights,advantages, andlogprobs.Type: String to TensorData object map
Required: No
See Also
For more information about using this API in one of the language-specific AWS SDKs, see the following:
