Skip to content

OperationConfig module-attribute 

OperationConfig: TypeAlias = (
    OperationConfigForwardMember
    | OperationConfigForwardBackwardMember
    | OperationConfigOptimizerStepMember
    | OperationConfigSaveWeightsMember
    | OperationConfigSaveStateMember
    | OperationConfigLoadStateMember
    | UnknownUnionMember
)
Used by

OperationConfigForwardMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
22
23
class OperationConfigForwardMember(TypedDict, closed=True):
    forward: "capo_sagemakertrainingsessionruntime.types.forward_config.ForwardConfig"

forward instance-attribute 

forward: ForwardConfig

OperationConfigForwardBackwardMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
26
27
class OperationConfigForwardBackwardMember(TypedDict, closed=True):
    forwardBackward: "capo_sagemakertrainingsessionruntime.types.forward_backward_config.ForwardBackwardConfig"

forwardBackward instance-attribute 

forwardBackward: ForwardBackwardConfig

OperationConfigOptimizerStepMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
30
31
class OperationConfigOptimizerStepMember(TypedDict, closed=True):
    optimizerStep: "capo_sagemakertrainingsessionruntime.types.optimizer_step_config.OptimizerStepConfig"

optimizerStep instance-attribute 

optimizerStep: OptimizerStepConfig

OperationConfigSaveWeightsMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
34
35
class OperationConfigSaveWeightsMember(TypedDict, closed=True):
    saveWeights: "capo_sagemakertrainingsessionruntime.types.save_weights_config.SaveWeightsConfig"

saveWeights instance-attribute 

saveWeights: SaveWeightsConfig

OperationConfigSaveStateMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
38
39
40
41
class OperationConfigSaveStateMember(TypedDict, closed=True):
    saveState: (
        "capo_sagemakertrainingsessionruntime.types.save_state_config.SaveStateConfig"
    )

saveState instance-attribute 

saveState: SaveStateConfig

OperationConfigLoadStateMember

Used by
Source code in docs_src/capo_sagemakertrainingsessionruntime/types/operation_config.py 
44
45
46
47
class OperationConfigLoadStateMember(TypedDict, closed=True):
    loadState: (
        "capo_sagemakertrainingsessionruntime.types.load_state_config.LoadStateConfig"
    )

loadState instance-attribute 

loadState: LoadStateConfig