Lines Matching defs:Optimizer

1620     def _state_dict_pre_hook(optimizer: Optimizer) -> None:
1625         optimizer: Optimizer, state_dict: Dict[str, Any]
1677         optimizer: Optimizer, state_dict: Dict[str, Any]
1683         optimizer: Optimizer, state_dict: Dict[str, Any]
1692     def _load_state_dict_post_hook(optimizer: Optimizer) -> None:
1762         def post_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1794         def pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1826         def global_pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1830         def global_post_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1834         def local_pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1838         def local_post_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):