OperationConfig
The configuration for a training operation. Exactly one member must be set to specify the operation type and its parameters.
Contents
Important
This data type is a UNION, so only one of the following members can be specified when used or returned.
- forward
-
The configuration for a forward-only pass through the model without gradient computation.
Type: ForwardConfig object
Required: No
- forwardBackward
-
The configuration for a forward and backward pass that computes gradients.
Type: ForwardBackwardConfig object
Required: No
- loadState
-
The configuration for loading training state from a previously saved checkpoint.
Type: LoadStateConfig object
Required: No
- optimizerStep
-
The configuration for an optimizer step that updates model weights using accumulated gradients.
Type: OptimizerStepConfig object
Required: No
- saveState
-
The configuration for saving full training state, including model weights and optimizer state.
Type: SaveStateConfig object
Required: No
- saveWeights
-
The configuration for saving model weights as a checkpoint.
Type: SaveWeightsConfig object
Required: No
See Also
For more information about using this API in one of the language-specific Amazon SDKs, see the following: