MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / _get_optimizer_param

Method _get_optimizer_param

deepspeed/runtime/engine.py:3228–3237  ·  view source on GitHub ↗
(self, param_name)

Source from the content-addressed store, hash-verified

3226 self.monitor.write_events(self.summary_events)
3227
3228 def _get_optimizer_param(self, param_name):
3229 result = []
3230 if not self.optimizer:
3231 return result
3232 for group in self.optimizer.param_groups:
3233 if param_name in group:
3234 result.append(group[param_name])
3235 else:
3236 result.append(0.0)
3237 return result
3238
3239 def _get_optimizer_loss_scale(self):
3240 if not self.optimizer:

Callers 3

get_lrMethod · 0.95
get_typeMethod · 0.95
get_momMethod · 0.95

Calls 1

appendMethod · 0.80

Tested by

no test coverage detected