Searched defs:StateDict (Results 1 – 1 of 1) sorted by relevance
42 StateDict: TypeAlias = Dict[str, Any] variable605 hook: Callable[["Optimizer", StateDict], Optional[StateDict]], argument759 hook: Callable[["Optimizer", StateDict], Optional[StateDict]], argument831 def load_state_dict(self, state_dict: StateDict) -> None: