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

Method _get_optimizer_param

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

Source from the content-addressed store, hash-verified

3560 self.monitor.write_events(self.summary_events)
3561
3562 def _get_optimizer_param(self, param_name):
3563 result = []
3564 if not self.optimizer:
3565 return result
3566 for group in self.optimizer.param_groups:
3567 if param_name in group:
3568 result.append(group[param_name])
3569 else:
3570 result.append(0.0)
3571 return result
3572
3573 def _get_optimizer_loss_scale(self):
3574 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