Searched defs:add_param_group (Results 1 – 8 of 8) sorted by relevance
109 def add_param_group(self, param_group): member in PostLocalSGDOptimizer
285 def add_param_group(self, param_group: Mapping[str, Any]) -> None: member in _NamedOptimizer
459 def add_param_group(self, param_group: Dict[str, Any]) -> None: member in ZeroRedundancyOptimizer
95 def add_param_group(self, param_group: Any): member in ShardedOptimizer
49 void SGD::add_param_group(const SGDParamGroup& param_group) { in add_param_group() function in torch::jit::mobile::SGD
98 void Optimizer::add_param_group(const OptimizerParamGroup& param_group) { in add_param_group() function in torch::optim::Optimizer
74 void SGD::add_param_group(const SGDParamGroup& param_group) { in add_param_group() function in executorch::extension::training::optimizer::SGD
993 def add_param_group(self, param_group: Dict[str, Any]) -> None: member in Optimizer