Searched defs:OptimizerStateType (Results 1 – 1 of 1) sorted by relevance
87 OptimizerStateType = Dict[str, Union[DictValueType, ListDictValueType]] variable396 optim_state_dict: OptimizerStateType,624 def _flatten_optim_state_dict(state_dict: OptimizerStateType) -> Dict[str, ValueType]:800 optim_state_dict: OptimizerStateType,873 state_dict: OptimizerStateType,1209 optim_state_dict: OptimizerStateType,1250 optim_state_dict: OptimizerStateType,