Interface OperationConfig.Builder

All Superinterfaces:

Buildable, CopyableBuilder<OperationConfig.Builder,OperationConfig>, SdkBuilder<OperationConfig.Builder,OperationConfig>, SdkPojo

Enclosing class:

OperationConfig

@Mutable
@NotThreadSafe
public static interface OperationConfig.Builder
extends SdkPojo, CopyableBuilder<OperationConfig.Builder,OperationConfig>

Method Summary

Modifier and Type	Method	Description
default OperationConfig.Builder	forward(Consumer<ForwardConfig.Builder> forward)	The configuration for a forward-only pass through the model without gradient computation.
OperationConfig.Builder	forward(ForwardConfig forward)	The configuration for a forward-only pass through the model without gradient computation.
default OperationConfig.Builder	forwardBackward(Consumer<ForwardBackwardConfig.Builder> forwardBackward)	The configuration for a forward and backward pass that computes gradients.
OperationConfig.Builder	forwardBackward(ForwardBackwardConfig forwardBackward)	The configuration for a forward and backward pass that computes gradients.
default OperationConfig.Builder	loadState(Consumer<LoadStateConfig.Builder> loadState)	The configuration for loading training state from a previously saved checkpoint.
OperationConfig.Builder	loadState(LoadStateConfig loadState)	The configuration for loading training state from a previously saved checkpoint.
default OperationConfig.Builder	optimizerStep(Consumer<OptimizerStepConfig.Builder> optimizerStep)	The configuration for an optimizer step that updates model weights using accumulated gradients.
OperationConfig.Builder	optimizerStep(OptimizerStepConfig optimizerStep)	The configuration for an optimizer step that updates model weights using accumulated gradients.
default OperationConfig.Builder	saveState(Consumer<SaveStateConfig.Builder> saveState)	The configuration for saving full training state, including model weights and optimizer state.
OperationConfig.Builder	saveState(SaveStateConfig saveState)	The configuration for saving full training state, including model weights and optimizer state.
default OperationConfig.Builder	saveWeights(Consumer<SaveWeightsConfig.Builder> saveWeights)	The configuration for saving model weights as a checkpoint.
OperationConfig.Builder	saveWeights(SaveWeightsConfig saveWeights)	The configuration for saving model weights as a checkpoint.

Methods inherited from interface software.amazon.awssdk.utils.builder.CopyableBuilder

copy

Methods inherited from interface software.amazon.awssdk.utils.builder.SdkBuilder

applyMutation, build

Methods inherited from interface software.amazon.awssdk.core.SdkPojo

equalsBySdkFields, sdkFieldNameToField, sdkFields

Method Details

forward

OperationConfig.Builder forward(ForwardConfig forward)

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

Parameters:

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

Returns:

Returns a reference to this object so that method calls can be chained together.

forward

default OperationConfig.Builder forward(Consumer<ForwardConfig.Builder> forward)

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

This is a convenience method that creates an instance of the ForwardConfig.Builder avoiding the need to create one manually via ForwardConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to forward(ForwardConfig).

Parameters:

forward - a consumer that will call methods on ForwardConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• forward(ForwardConfig)

forwardBackward

OperationConfig.Builder forwardBackward(ForwardBackwardConfig forwardBackward)

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

Parameters:

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

Returns:

Returns a reference to this object so that method calls can be chained together.

forwardBackward

default OperationConfig.Builder forwardBackward(Consumer<ForwardBackwardConfig.Builder> forwardBackward)

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

This is a convenience method that creates an instance of the ForwardBackwardConfig.Builder avoiding the need to create one manually via ForwardBackwardConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to forwardBackward(ForwardBackwardConfig).

Parameters:

forwardBackward - a consumer that will call methods on ForwardBackwardConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• forwardBackward(ForwardBackwardConfig)

optimizerStep

OperationConfig.Builder optimizerStep(OptimizerStepConfig optimizerStep)

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

Parameters:

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

Returns:

Returns a reference to this object so that method calls can be chained together.

optimizerStep

default OperationConfig.Builder optimizerStep(Consumer<OptimizerStepConfig.Builder> optimizerStep)

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

This is a convenience method that creates an instance of the OptimizerStepConfig.Builder avoiding the need to create one manually via OptimizerStepConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to optimizerStep(OptimizerStepConfig).

Parameters:

optimizerStep - a consumer that will call methods on OptimizerStepConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• optimizerStep(OptimizerStepConfig)

saveWeights

OperationConfig.Builder saveWeights(SaveWeightsConfig saveWeights)

The configuration for saving model weights as a checkpoint.

Parameters:

saveWeights - The configuration for saving model weights as a checkpoint.

Returns:

Returns a reference to this object so that method calls can be chained together.

saveWeights

default OperationConfig.Builder saveWeights(Consumer<SaveWeightsConfig.Builder> saveWeights)

The configuration for saving model weights as a checkpoint.

This is a convenience method that creates an instance of the SaveWeightsConfig.Builder avoiding the need to create one manually via SaveWeightsConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to saveWeights(SaveWeightsConfig).

Parameters:

saveWeights - a consumer that will call methods on SaveWeightsConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• saveWeights(SaveWeightsConfig)

saveState

OperationConfig.Builder saveState(SaveStateConfig saveState)

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

Parameters:

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

Returns:

Returns a reference to this object so that method calls can be chained together.

saveState

default OperationConfig.Builder saveState(Consumer<SaveStateConfig.Builder> saveState)

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

This is a convenience method that creates an instance of the SaveStateConfig.Builder avoiding the need to create one manually via SaveStateConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to saveState(SaveStateConfig).

Parameters:

saveState - a consumer that will call methods on SaveStateConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• saveState(SaveStateConfig)

loadState

OperationConfig.Builder loadState(LoadStateConfig loadState)

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

Parameters:

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

Returns:

Returns a reference to this object so that method calls can be chained together.

loadState

default OperationConfig.Builder loadState(Consumer<LoadStateConfig.Builder> loadState)

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

This is a convenience method that creates an instance of the LoadStateConfig.Builder avoiding the need to create one manually via LoadStateConfig.builder().

When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to loadState(LoadStateConfig).

Parameters:

loadState - a consumer that will call methods on LoadStateConfig.Builder

Returns:

Returns a reference to this object so that method calls can be chained together.

See Also:

• loadState(LoadStateConfig)